Added the (missing) W_o

[mygpt.git] / main.py

diff --git a/main.py b/main.py
index a18beb1..3bf6b52 100755 (executable)
--- a/main.py
+++ b/main.py
@@ -131,8 +131,8 @@ class TaskPicoCLVR(Task):
             many_colors = many_colors
         )
 
-        self.test_descr = descr[:nb // 5]
-        self.train_descr = descr[nb // 5:]
+        # self.test_descr = descr[:nb // 5]
+        # self.train_descr = descr[nb // 5:]
 
         descr = [ s.strip().split(' ') for s in descr ]
         l = max([ len(s) for s in descr ])
@@ -201,7 +201,8 @@ class TaskPicoCLVR(Task):
                                      file_name, nrow = nb_per_primer, pad_value = 0.8)
         log_string(f'wrote {file_name}')
 
-        log_string(f'nb_misssing {picoclvr.nb_missing_properties(descr)}')
+        nb_missing = sum( [ x[2] for x in picoclvr.nb_missing_properties(descr) ] )
+        log_string(f'nb_missing {nb_missing / len(descr):.02f}')
 
 ######################################################################