mirror of
https://github.com/deepseek-ai/DeepSeek-Coder.git
synced 2025-04-19 18:19:05 -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()
|
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
|
from model import IrisClassifier as Classifier
|
||||||
|
|
||||||
def pipeline():
|
def main():
|
||||||
# Model training and evaluation
|
# Model training and evaluation
|
||||||
"""
|
"""
|
||||||
inputs = tokenizer.encode(input_text, return_tensors="pt").to(device)
|
inputs = tokenizer.encode(input_text, return_tensors="pt").to(device)
|
||||||
|
Loading…
Reference in New Issue
Block a user