I think you might need to reset the index after dropna()
dropna()
trainingData = trainingData.dropna() trainingData.reset_index(drop=True, inplace=True) trainingData['text'] = trainingData['text'].apply(clean_tweet) ...