From a1ebd1b7260e56f75b4cd94ed9fefea240f360d0 Mon Sep 17 00:00:00 2001 From: Alexandre Drouin Date: Tue, 30 May 2017 17:16:19 -0400 Subject: [PATCH] Changed loss to new keyword in tests --- mmit/tests/test_learner.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mmit/tests/test_learner.py b/mmit/tests/test_learner.py index 766670f..5cf476f 100644 --- a/mmit/tests/test_learner.py +++ b/mmit/tests/test_learner.py @@ -74,7 +74,7 @@ def test_noiseless_dataset_1_linear(self): np.arange(len(y)).reshape(-1, 1), # Good feature np.zeros(len(y)).reshape(-1, 1))) # Bad feature - estimator = MaxMarginIntervalTree(margin=0.0, loss="hinge") + estimator = MaxMarginIntervalTree(margin=0.0, loss="linear_hinge") estimator.fit(X, y) assert len(estimator.tree_.rules) == 1