fix class value

kermitt2 · Aug 28, 2023 · e22c86c · e22c86c
1 parent 2f8976c
commit e22c86c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/delft/textClassification/reader.py b/delft/textClassification/reader.py
@@ -234,10 +234,10 @@ def load_dataseer_corpus_csv(filepath):
         reuse_list = np.asarray(reuse_list)
         # map boolean values to [0,1]
         def map_boolean(x):
-            return [1.0,0.0] if x else [0.0,1.0]
+            return [1.0,0.0] if x == 'no_reuse' else [0.0,1.0]
         reuse_list = np.array(list(map(map_boolean, reuse_list)))
         print(reuse_list)
-        return np.asarray(texts_list), reuse_list, None, None, ["not_reuse", "reuse"], None, None
+        return np.asarray(texts_list), reuse_list, None, None, ["no_reuse", "reuse"], None, None
 
     # otherwise we have the list of datatypes, and optionally subtypes and leaf datatypes
     datatypes = df.iloc[:,2]