【随机森林】Python以热力图形式展示图片各像元的特征重要性

hi_ly_51

已于 2024-01-16 11:29:13 修改

阅读量401

点赞数 11

文章标签： python numpy 开发语言

于 2024-01-15 17:45:52 首次发布

本文链接：https://blog.csdn.net/weixin_62100318/article/details/135607304

版权

对手写数字数据集进行随机森林集成分析

import matplotlib
import matplotlib.pyplot as plt
from sklearn.ensemble import RandomForestClassifier

'''导入手写数字数据集'''
from sklearn.datasets import fetch_openml

mnist = fetch_openml("mnist_784")
# print(mnist)

rf_clf = RandomForestClassifier(n_estimators=500, n_jobs=-1)
rf_clf.fit(mnist["data"], mnist["target"])

'''绘制热力图'''


def plot_digit(data):
    image = data.reshape(28, 28)
    plt.imshow(image, cmap=matplotlib.cm.hot)
    plt.axis('off')


plot_digit(rf_clf.feature_importances_)
char = plt.colorbar(ticks=[rf_clf.feature_importances_.min(), rf_clf.feature_importances_.max()])
char.ax.set_yticklabels(["Not important", "Very important"])
plt.show()

最终得到的结果：