mirror of
https://github.com/deepseek-ai/DeepSeek-Coder.git
synced 2025-04-19 10:09:14 -04:00
Update README.md
This commit is contained in:
parent
8ba4cdcb63
commit
e3c9b42f86
@ -135,9 +135,9 @@ class IrisClassifier(nn.Module):
|
||||
return predicted.numpy()
|
||||
#main.py
|
||||
from utils import load_data, evaluate_predictions
|
||||
from model import IrisClassifier
|
||||
from model import IrisClassifier as Classifier
|
||||
|
||||
def pipeline():
|
||||
def main():
|
||||
# Model training and evaluation
|
||||
"""
|
||||
inputs = tokenizer.encode(input_text, return_tensors="pt").to(device)
|
||||
|
Loading…
Reference in New Issue
Block a user