5、用RFE,对像素进行排序
from sklearn.svm import SVC
from sklearn.datasets import load_digits
from sklearn.feature_selection import RFE
import matplotlib.pyplot as plt
plt.rcParams['font.sans-serif'] = ['SimHei']
plt.rcParams['axes.unicode_minus'] = False
# 加载数字数据集
digits = load_digits()
X = digits.images.reshape((len(digits.images), -1))
y = digits.target
# 创建RFE对象并对每个像素进行排序
svc = SVC(kernel="linear", C=1)
rfe = RFE(estimator=svc, n_features_to_select=1, step=1)
rfe.fit(X, y)
ranking = rfe.ranking_.reshape(digits.images[0].shape)
# 图像素排序
plt.matshow(ranking, cmap=plt.cm.Greens)
plt.colorbar()
plt.title("用RFE,对像素进行排序")
plt.show()