BecomeAllan
/

ML-SLRC

Model card Files Files and versions

Ssddsasd commited on May 21, 2022

Commit

4bb8faf

·

1 Parent(s): 1ba9831

Update Util_funs.py

Files changed (1) hide show

Util_funs.py +1 -1

Util_funs.py CHANGED Viewed

@@ -76,7 +76,7 @@ def prepare_data(data, batch_size,tokenizer,max_seq_length,
   rest_labaled_data = labaled_data.loc[~labaled_data.index.isin(data_train.index),:]
   unlabaled_data = data.loc[data['label'].isna()]
-  data_test = pd.concat([rest_labaled_data, unlabaled_data])
   # Train

   rest_labaled_data = labaled_data.loc[~labaled_data.index.isin(data_train.index),:]
   unlabaled_data = data.loc[data['label'].isna()]
+  data_test=data
   # Train