from sklearn.linear_model import LogisticRegression | |
# LR model | |
model_bert = LogisticRegression() | |
# train | |
model_bert = model_bert.fit(X_tr_bert, y_tr) | |
# predict | |
pred_bert = model_bert.predict(X_val_bert) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment