EPGCN-Tests

benedekrozemberczki · benedekrozemberczki · commit 4a1d70124ee3 · 2022-01-13T18:32:03.000Z
diff --git a/tests/unit/test_models.py b/tests/unit/test_models.py
@@ -26,7 +26,7 @@ def setUp(self):
         drug_feature_set = loader.get_drug_features()
         context_feature_set = loader.get_context_features()
         labeled_triples = loader.get_labeled_triples()
-        labeled_triples, _ = labeled_triples.train_test_split(train_size=0.02)
+        labeled_triples, _ = labeled_triples.train_test_split(train_size=0.005)
         self.generator = BatchGenerator(
             batch_size=32, context_features=True, drug_features=True, drug_molecules=True, labels=True
         )

Original file line number	Diff line number	Diff line change
`@@ -26,7 +26,7 @@ def setUp(self):`
`26`	`26`	`drug_feature_set = loader.get_drug_features()`
`27`	`27`	`context_feature_set = loader.get_context_features()`
`28`	`28`	`labeled_triples = loader.get_labeled_triples()`
`29`		`- labeled_triples, _ = labeled_triples.train_test_split(train_size=0.02)`
	`29`	`+ labeled_triples, _ = labeled_triples.train_test_split(train_size=0.005)`
`30`	`30`	`self.generator = BatchGenerator(`
`31`	`31`	`batch_size=32, context_features=True, drug_features=True, drug_molecules=True, labels=True`
`32`	`32`	`)`