We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 4fa8262 commit 792885eCopy full SHA for 792885e
1 file changed
test/classifier_test.rb
@@ -31,8 +31,7 @@ def test_multiclass
31
model.fit(x_train, y_train)
32
y_pred = model.predict(x_test)
33
expected = [1, 2, 2, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 2, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 2, 2, 1, 1, 1, 0, 1, 2, 1, 2, 1, 2, 1, 1, 1, 1, 1, 1, 1, 1, 2, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 2, 1, 1, 1, 1, 1, 1, 1, 1, 2, 2, 1, 1, 1, 2, 1, 1, 1, 1, 1, 1, 1]
34
- # TODO fix
35
- # assert_equal expected, y_pred.first(100)
+ assert_equal expected, y_pred.first(100)
36
37
y_pred_proba = model.predict_proba(x_test)
38
expected = [0.0020083063282072544, 0.9485360383987427, 0.04945564642548561]
0 commit comments