Spaces:
Runtime error
Runtime error
File size: 835 Bytes
3065bfd 559e539 525392a 559e539 348106a ede03d8 2340f72 348106a c165949 655b400 348106a f0cf6b4 c97369f 94a4ce2 f0cf6b4 cc827c8 94a4ce2 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 | # Utilities
import sys
import torch
sys.modules['torch'] = torch
import os
os.environ["CUDA_VISIBLE_DEVICES"] = ""
from src.modello import Modello
from src.dataset import LoadDataset
from sklearn.metrics import accuracy_score
class TestClass :
"""
unit test sul modello per pipeline CI
"""
def test_trivial_output(self) :
# Controllo del funzionamento del modello con frasi banali
model = Modello()
assert model.predict("neutral")[0]=="neutral" and model.predict("awesome")[0]=="positive" and model.predict("terrible")[0]=="negative"
def test_accuracy(self) :
# Controllo che l'accuracy sia almeno 0.5
model = Modello()
ld = LoadDataset()
X = ld.X
y = ld.y
y_pred = model.predict(X)
assert accuracy_score(y, y_pred)>=0.5
|