Merge pull request #160 from YaoZh1918/master

fix bug in cross_validate ret['train_' + m] was returning testset values
NicolasHug · Mar 30, 2018 · 56e534a · 56e534a
2 parents 58f3508 + ff03506
commit 56e534a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/surprise/model_selection/validation.py b/surprise/model_selection/validation.py
@@ -116,7 +116,7 @@ def cross_validate(algo, data, measures=['rmse', 'mae'], cv=None,
         if return_train_measures:
             train_measures[m] = np.asarray([d[m] for d in
                                             train_measures_dicts])
-            ret['train_' + m] = test_measures[m]
+            ret['train_' + m] = train_measures[m]
 
     ret['fit_time'] = fit_times
     ret['test_time'] = test_times