Skip to content

Instantly share code, notes, and snippets.

@mohdsanadzakirizvi
Created September 23, 2019 18:05
Show Gist options
  • Save mohdsanadzakirizvi/a535422980df13c47f4e28d548ae86d3 to your computer and use it in GitHub Desktop.
Save mohdsanadzakirizvi/a535422980df13c47f4e28d548ae86d3 to your computer and use it in GitHub Desktop.
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