Skip to content

Instantly share code, notes, and snippets.

@prateekjoshi565

prateekjoshi565/prediction_bert.py

Created July 18, 2020 11:19
Show Gist options
  • Save prateekjoshi565/ed8efd3f1210436c4733ab0a7db4d85f to your computer and use it in GitHub Desktop.
Save prateekjoshi565/ed8efd3f1210436c4733ab0a7db4d85f to your computer and use it in GitHub Desktop.
Download ZIP
Raw
prediction_bert.py
# get predictions for test data
with torch.no_grad():
preds = model(test_seq.to(device), test_mask.to(device))
preds = preds.detach().cpu().numpy()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment