Updated validation set size

PreferredAI · Oct 17, 2023 · 6bad38f · 6bad38f
1 parent a87d9bf
commit 6bad38f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/ngcf_example.py b/examples/ngcf_example.py
@@ -25,8 +25,8 @@
 # Instantiate an evaluation method to split data into train and test sets.
 ratio_split = RatioSplit(
     data=data,
-    # val_size=0.1,
-    test_size=0.2,
+    val_size=0.1,
+    test_size=0.1,
     exclude_unknowns=True,
     verbose=True,
     seed=123,