擅长:python、mysql、java
<p>下面的代码将生成十大功能的图表:</p>
<pre><code>import numpy as np
import matplotlib.pyplot as plt
importances = clf.feature_importances_
std = np.std(clf.feature_importances_,axis=0)
indices = np.argsort(importances)[::-1]
# Print the feature ranking
print("Feature ranking:")
for f in range(10):
print("%d. feature %d (%f)" % (f + 1, indices[f], importances[indices[f]]))
# Plot the feature importances of the forest
plt.figure()
plt.title("Feature importances")
plt.bar(range(10), importances[indices],
color="r", yerr=std[indices], align="center")
plt.xticks(range(10), indices)
plt.xlim([-1, 10])
plt.show()
</code></pre>
<p>取自<a href="http://scikit-learn.org/stable/auto_examples/ensemble/plot_forest_importances.html" rel="nofollow">here</a>并稍加修改以适应<a href="http://scikit-learn.org/stable/modules/generated/sklearn.tree.DecisionTreeClassifier.html" rel="nofollow">DecisionTreeClassifier</a>。</p>
<p>这并不完全有助于你探索这棵树,但它确实告诉你关于这棵树的事情。</p>