Skip to content

Instantly share code, notes, and snippets.

@WillKoehrsen

WillKoehrsen/predict_rf.py

Last active August 30, 2018 16:53
Show Gist options
  • Save WillKoehrsen/46237dbd5df103c06db998bae28086bf to your computer and use it in GitHub Desktop.
Save WillKoehrsen/46237dbd5df103c06db998bae28086bf to your computer and use it in GitHub Desktop.
Download ZIP
Raw
predict_rf.py
# Actual class predictions
rf_predictions = model.predict(test)
# Probabilities for each class
rf_probs = model.predict_proba(test)[:, 1]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment