deepseekmirror/DeepSeek-Coder
1
0
Fork 0
mirror of https://github.com/deepseek-ai/DeepSeek-Coder.git synced 2025-02-23 06:09:07 -05: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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 0 deletions
Show Stats 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