Skip to content

Instantly share code, notes, and snippets.

@spikar

spikar/feature_importance.py

Created May 24, 2019 10:21
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 spikar/8d44cf8b732f17cd6bf081942a40196a to your computer and use it in GitHub Desktop.
Save spikar/8d44cf8b732f17cd6bf081942a40196a to your computer and use it in GitHub Desktop.
Download ZIP
Raw
feature_importance.py
import pandas as pd
import numpy as np
from sklearn.ensemble import ExtraTreesClassifier
import matplotlib.pyplot as plt
data = pd.read_csv("D://Blogs//train.csv")
model = ExtraTreesClassifier()
model.fit(X,y)
print(model.feature_importances_) #use inbuilt class feature_importances of tree based classifiers
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment