Allan Victor commited on
Commit
fac590f
·
1 Parent(s): c2b793e

Update Util_funs.py

Browse files
Files changed (1) hide show
  1. Util_funs.py +1 -1
Util_funs.py CHANGED
@@ -76,7 +76,7 @@ def prepare_data(data, batch_size, tokenizer,max_seq_length,
76
  idex = data.index.isin(data_train.index)
77
 
78
  # The Test set to label by the model
79
- data_test = data[~idex].reset_index()
80
 
81
 
82
  # Transform in dataset to model
 
76
  idex = data.index.isin(data_train.index)
77
 
78
  # The Test set to label by the model
79
+ data_test = data
80
 
81
 
82
  # Transform in dataset to model