Mercurial > hg > plosone_underreview

diff tests/test_classification.py @ 93:f9513664fe42 branch-tests

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
notebooks update
author mpanteli <m.x.panteli@gmail.com>
date Mon, 02 Oct 2017 18:58:39 +0100
parents d118b6ca8370
children
line wrap: on
line diff
--- a/tests/test_classification.py	Mon Oct 02 15:34:20 2017 +0100
+++ b/tests/test_classification.py	Mon Oct 02 18:58:39 2017 +0100
@@ -19,7 +19,7 @@
     X[-50:, :] = X[-50:, :] + 10
     Y = np.concatenate([np.repeat('a', 50), np.repeat('b', 50)])
     X_train, X_test, Y_train, Y_test = train_test_split(X, Y, train_size=0.6, random_state=1, stratify=Y)
-    accuracy, _ = classification.confusion_matrix(X_train, Y_train, X_test, Y_test)
+    accuracy, _, _ = classification.confusion_matrix(X_train, Y_train, X_test, Y_test)
     # expect perfect accuracy for this 'easy' dataset
     assert accuracy == 1.0