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