/bert_model.py
Created Sep 23, 2019
| 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