diff --git a/train-tensorflow.py b/train-tensorflow.py index 3a6a3c3..bfccaff 100644 --- a/train-tensorflow.py +++ b/train-tensorflow.py @@ -23,9 +23,9 @@ test_df = pd.read_csv('test.csv') df.dropna(subset = ['reviews.text'], inplace = True) -train_df.dropna(subset = ['reviews.text'], inplace = True) val_df.dropna(subset = ['reviews.text'], inplace = True) test_df.dropna(subset = ['reviews.text'], inplace = True) +train_df.dropna(subset = ['reviews.text'], inplace = True) train_sentences = train_df['reviews.text'].to_numpy()