Ejemplo n.º 1
0
 def test_predict_proba_to_label_unknown(self):
     testable = SKLinearImageModel(pkl_file=None)
     proba = (0.45, 0.65)
     self.assertEqual('unknown', testable._predict_proba_to_label(proba))
Ejemplo n.º 2
0
 def test_predict_proba_to_label_dog(self):
     testable = SKLinearImageModel(pkl_file=None)
     proba = (0.15, 0.85)
     self.assertEqual('dog', testable._predict_proba_to_label(proba))