new RuleBasedDataset and EnviFormer dataset working for respective models #120

2025-11-04 10:58:16 +13:00
parent ff51e48f90
commit ac5d370b18
5 changed files with 126 additions and 101 deletions
--- a/tests/test_enviformer.py
+++ b/tests/test_enviformer.py
@ -50,7 +50,7 @@ class EnviFormerTest(TestCase):
                mod.build_model()
                mod.multigen_eval = True
                mod.save()
-                mod.evaluate_model()
+                mod.evaluate_model(n_splits=2)

                mod.predict("CCN(CC)C(=O)C1=CC(=CC=C1)C")