From e1cc9ae6d90ca6775f8488c92a4f42b37a055fb9 Mon Sep 17 00:00:00 2001 From: Chenggang Zhao Date: Mon, 6 Nov 2023 00:27:10 +0800 Subject: [PATCH] Update README.md --- README.md | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index 3044b0f..cf02127 100644 --- a/README.md +++ b/README.md @@ -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)