DeepSeek/DeepSeek-Coder
1
0
Fork 0
mirror of https://github.com/deepseek-ai/DeepSeek-Coder synced 2025-06-26 18:25:53 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update README.md

Browse Source
This commit is contained in:
Chenggang Zhao
2023-11-06 00:29:00 +08:00
committed by GitHub
parent e1cc9ae6d9
commit e593c64f7b
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
README.md
View File

@@ -203,6 +203,8 @@ def load_data():
def evaluate_predictions(y_test, y_pred): def evaluate_predictions(y_test, y_pred):
return accuracy_score(y_test, y_pred) return accuracy_score(y_test, y_pred)
# model.py # model.py
import torch import torch
import torch.nn as nn import torch.nn as nn
@@ -242,6 +244,8 @@ class IrisClassifier(nn.Module):
outputs = self(X_test) outputs = self(X_test)
_, predicted = outputs.max(1) _, predicted = outputs.max(1)
return predicted.numpy() return predicted.numpy()
# main.py # main.py
from utils import load_data, evaluate_predictions from utils import load_data, evaluate_predictions
from model import IrisClassifier as Classifier from model import IrisClassifier as Classifier