fix bug in cross_validate

fix a bug in cross_validate when return_train_measures=True
jiapeijia · Mar 30, 2018 · ff03506 · ff03506
1 parent 58f3508
commit ff03506
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