diff --git a/tests/cornac/eval_methods/test_propensity_stratified_evaluation.py b/tests/cornac/eval_methods/test_propensity_stratified_evaluation.py index 6d61a0d483751a738dec8491b56bfe2fc1379f65..c6d4c3a37f258c0c494c66450f5f12dd99d43624 100644 --- a/tests/cornac/eval_methods/test_propensity_stratified_evaluation.py +++ b/tests/cornac/eval_methods/test_propensity_stratified_evaluation.py @@ -27,7 +27,7 @@ from cornac.metrics import MAE, AUC class TestPropensityStratifiedEvaluation(unittest.TestCase): def setUp(self): self.ml_100k = cornac.datasets.movielens.load_feedback() - cutoff = int(len(self.ml_100k) / 100) # use 1% for faster testing + cutoff = int(len(self.ml_100k) * 0.1) # use 10% for faster testing self.ml_100k = self.ml_100k[:cutoff] def test_stratified_split(self, n_strata=2):