From 937b5f038d4a04a2affe50e53290d2db5aec0e9a Mon Sep 17 00:00:00 2001 From: Quoc-Tuan Truong <tqtg@users.noreply.github.com> Date: Thu, 8 Jul 2021 16:24:37 +0800 Subject: [PATCH] Increase sample size to avoid empty test set (#416) --- .../eval_methods/test_propensity_stratified_evaluation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/cornac/eval_methods/test_propensity_stratified_evaluation.py b/tests/cornac/eval_methods/test_propensity_stratified_evaluation.py index 6d61a0d4..c6d4c3a3 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): -- GitLab