DeepSeek/DeepSeek-Coder
1
0
Fork 0
mirror of https://github.com/deepseek-ai/DeepSeek-Coder synced 2024-12-04 18:14:44 +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:27:10 +08:00 committed by GitHub
parent 33c230697b
commit e1cc9ae6d9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
README.md
View File

@ -203,7 +203,7 @@ def load_data():
def evaluate_predictions(y_test, y_pred):
return accuracy_score(y_test, y_pred)
#model.py
# model.py
import torch
import torch.nn as nn
import torch.optim as optim
@ -242,7 +242,7 @@ class IrisClassifier(nn.Module):
outputs = self(X_test)
_, predicted = outputs.max(1)
return predicted.numpy()
#main.py
# main.py
from utils import load_data, evaluate_predictions
from model import IrisClassifier as Classifier
@ -255,7 +255,7 @@ print(tokenizer.decode(outputs[0]))
```
---
In the following scenario, the DeepSeek-Coder 6.7B model effectively calls a class **IrisClassifier** and its member function from the `model.py` file, and also utilizes functions from the `utils.py` file, to correctly complete the **main** function in`main.py` file for model training and evaluation.
In the following scenario, the DeepSeek-Coder-6.7B model effectively calls a class **IrisClassifier** and its member function from the `model.py` file, and also utilizes functions from the `utils.py` file, to correctly complete the **main** function in `main.py` file for model training and evaluation.
![Completion GIF](pictures/completion_demo.gif)