Skip to content

Instantly share code, notes, and snippets.

@mohdsanadzakirizvi

mohdsanadzakirizvi/bert_model.py

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.
Download ZIP
Raw
bert_model.py
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