Skip to content
Snippets Groups Projects
Commit 796f8d62 authored by bbauvin's avatar bbauvin
Browse files

Minor changes

parent 4b0b0485
No related branches found
No related tags found
No related merge requests found
...@@ -3,6 +3,7 @@ from sklearn.pipeline import Pipeline ...@@ -3,6 +3,7 @@ from sklearn.pipeline import Pipeline
from sklearn.model_selection import RandomizedSearchCV from sklearn.model_selection import RandomizedSearchCV
from scipy.stats import randint from scipy.stats import randint
import numpy as np import numpy as np
import cPickle
from .. import Metrics from .. import Metrics
from ..utils.HyperParameterSearch import genHeatMaps from ..utils.HyperParameterSearch import genHeatMaps
...@@ -89,4 +90,18 @@ def getConfig(config): ...@@ -89,4 +90,18 @@ def getConfig(config):
def getInterpret(classifier, directory): def getInterpret(classifier, directory):
pass featureImportances = classifier.feature_importances_
sortedArgs = np.argsort(-featureImportances)
featureImportancesSorted = featureImportances[sortedArgs][:50]
featureIndicesSorted = sortedArgs[:50]
featuresImportancesDict = dict((featureIndex, featureImportance)
for featureIndex, featureImportance in enumerate(featureImportances)
if featureImportance != 0)
with open(directory+'-feature_importances.pickle', 'wb') as handle:
cPickle.dump(featuresImportancesDict, handle)
interpretString = "Feature importances : \n"
for featureIndex, featureImportance in zip(featureIndicesSorted, featureImportancesSorted):
if featureImportance>0:
interpretString+="- Feature index : "+str(featureIndex)+ \
", feature importance : "+str(featureImportance)+"\n"
return interpretString
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment