I train a neural network with my train_data. The class of the label is 1 or 2.
pred = pd.Series(model.predict(X_test_tok).tolist())
So that i