Skip to content
Snippets Groups Projects
Unverified Commit 937b5f03 authored by Quoc-Tuan Truong's avatar Quoc-Tuan Truong Committed by GitHub
Browse files

Increase sample size to avoid empty test set (#416)

parent 9119f208
No related branches found
No related tags found
No related merge requests found
...@@ -27,7 +27,7 @@ from cornac.metrics import MAE, AUC ...@@ -27,7 +27,7 @@ from cornac.metrics import MAE, AUC
class TestPropensityStratifiedEvaluation(unittest.TestCase): class TestPropensityStratifiedEvaluation(unittest.TestCase):
def setUp(self): def setUp(self):
self.ml_100k = cornac.datasets.movielens.load_feedback() 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] self.ml_100k = self.ml_100k[:cutoff]
def test_stratified_split(self, n_strata=2): def test_stratified_split(self, n_strata=2):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment