Roboproch commited on
Commit
f0cf6b4
·
1 Parent(s): 3065bfd

fix minor per improt torch

Browse files
Files changed (1) hide show
  1. CI/test_ci.py +3 -3
CI/test_ci.py CHANGED
@@ -4,9 +4,6 @@ from src.modello import Modello
4
  from src.dataset import LoadDataset
5
  from sklearn.metrics import accuracy_score
6
 
7
- model = Modello()
8
- ld = LoadDataset()
9
-
10
  class TestClass :
11
  """
12
  unit test sul modello per pipeline CI
@@ -14,10 +11,13 @@ class TestClass :
14
 
15
  def test_trivial_output(self) :
16
  # Controllo del funzionamento del modello con frasi banali
 
17
  assert model.predict("neutral")[0]=="neutral" and model.predict("awesome")[0]=="positive" and model.predict("terrible")[0]=="negative"
18
 
19
  def test_accuracy(self) :
20
  # Controllo che l'accuracy sia almeno 0.5
 
 
21
  X = ld.X
22
  y = ld.y
23
  y_pred = model.predict(X)
 
4
  from src.dataset import LoadDataset
5
  from sklearn.metrics import accuracy_score
6
 
 
 
 
7
  class TestClass :
8
  """
9
  unit test sul modello per pipeline CI
 
11
 
12
  def test_trivial_output(self) :
13
  # Controllo del funzionamento del modello con frasi banali
14
+ model = Modello()
15
  assert model.predict("neutral")[0]=="neutral" and model.predict("awesome")[0]=="positive" and model.predict("terrible")[0]=="negative"
16
 
17
  def test_accuracy(self) :
18
  # Controllo che l'accuracy sia almeno 0.5
19
+ model = Modello()
20
+ ld = LoadDataset()
21
  X = ld.X
22
  y = ld.y
23
  y_pred = model.predict(X)