From e593c64f7b3f49038d43f5f67fd01f994c2fc253 Mon Sep 17 00:00:00 2001 From: Chenggang Zhao Date: Mon, 6 Nov 2023 00:29:00 +0800 Subject: [PATCH] Update README.md --- README.md | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/README.md b/README.md index cf02127..e2ae554 100644 --- a/README.md +++ b/README.md @@ -203,6 +203,8 @@ def load_data(): def evaluate_predictions(y_test, y_pred): return accuracy_score(y_test, y_pred) + + # model.py import torch import torch.nn as nn @@ -242,6 +244,8 @@ class IrisClassifier(nn.Module): outputs = self(X_test) _, predicted = outputs.max(1) return predicted.numpy() + + # main.py from utils import load_data, evaluate_predictions from model import IrisClassifier as Classifier