Skip to content

Instantly share code, notes, and snippets.

@FelixChop

FelixChop/first_10_predictions_class1.py

Created June 1, 2018 09:48
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save FelixChop/4bfd08b9c3d6a6d852c23be2f5ef1de0 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save FelixChop/4bfd08b9c3d6a6d852c23be2f5ef1de0 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
first_10_predictions_class1.py
df = X_test.copy()
df['predictions'] = rf_model.predict_proba(X_test)
data_to_analyze = df.sort_values('predictions', ascending=False).head(10)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment