有没有办法可以将xLabel(预测标签)0, 1移到混淆矩阵的顶部?感谢任何帮助。
from sklearn.metrics import accuracy_scoreplt.figure(figsize=(6,4))sns.heatmap(cm_rf, annot=True, fmt="d")plt.title('随机森林树 \n准确率:{0:.3%}\n'.format(accuracy_score(test_y, predict_rf_y)))plt.ylabel('真实标签')plt.xlabel('预测标签')plt.show()
回答:
你要找的属性是ax.xaxis.set_ticks_position('top')
和ax.xaxis.set_label_position('top')
。例如:
f, ax = plt.subplots(figsize=(6, 4))sns.heatmap(np.random.randint(0, 3, size=(2, 2)), annot=True, fmt="d")ax.set_title('随机森林树 \n准确率:{format}\n', y=1.08)ax.set_ylabel('真实标签')ax.set_xlabel('预测标签')ax.xaxis.set_ticks_position('top')ax.xaxis.set_label_position('top')
希望这对你有帮助