Skip to content

Instantly share code, notes, and snippets.

@jonathanoheix

jonathanoheix/nlp15.py

Created December 18, 2018 09:53
Show Gist options
  • Save jonathanoheix/5c86d26f257b9b6ba32d4c8e2d707bba to your computer and use it in GitHub Desktop.
Save jonathanoheix/5c86d26f257b9b6ba32d4c8e2d707bba to your computer and use it in GitHub Desktop.
Download ZIP
Raw
nlp15.py
# train a random forest classifier
rf = RandomForestClassifier(n_estimators = 100, random_state = 42)
rf.fit(X_train, y_train)
# show feature importance
feature_importances_df = pd.DataFrame({"feature": features, "importance": rf.feature_importances_}).sort_values("importance", ascending = False)
feature_importances_df.head(20)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment