Skip to content

Instantly share code, notes, and snippets.

@DoctorLoop

DoctorLoop/ModelPrediction.py

Created August 4, 2020 11:00
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save DoctorLoop/41c31bf30339dc1b4b4c7e69a001b9bf to your computer and use it in GitHub Desktop.
Save DoctorLoop/41c31bf30339dc1b4b4c7e69a001b9bf to your computer and use it in GitHub Desktop.
Download ZIP
Raw
ModelPrediction.py
num_samples = 5000
predicted = []
for i in range(num_samples):
features = test_dataset.df.sample(n=3).drop('outcome',axis=1)
batch_predictions = predict(torch.tensor(features.values))
predicted.append(batch_predictions)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment