BecomeAllan
/

ML-SLRC

Model card Files Files and versions Community

Allan Victor commited on May 24, 2022

Commit

9c0c4aa

·

1 Parent(s): cf2df3b

Update Util_funs.py

Files changed (1) hide show

Util_funs.py +1 -1

Util_funs.py CHANGED Viewed

@@ -794,7 +794,7 @@ class SLR_DataSet(Dataset):
     comment_text = self.treat_text(comment_text)
     try:
-      labels = LABEL_MAP[example[self.LABEL_NAME]]
     except:
       labels = -1

     comment_text = self.treat_text(comment_text)
     try:
+      labels = LABEL_MAP[example[self.LABEL_NAME].lower()]
     except:
       labels = -1