将skplt.metrics.plot_roc中的图例改成中文
时间: 2024-03-03 15:51:41 浏览: 98
要将 `skplt.metrics.plot_roc` 中的图例改成中文,可以添加 `title_fontsize` 和 `text_fontsize` 参数,并设置为中文字体即可。以下是一个示例代码:
```python
import matplotlib.pyplot as plt
from sklearn.metrics import roc_curve, auc
import scikitplot as skplt
# 生成 ROC 曲线数据
y_prob = clf.predict_proba(X_test)
fpr, tpr, thresholds = roc_curve(y_test, y_prob[:, 1])
roc_auc = auc(fpr, tpr)
# 绘制 ROC 曲线
skplt.metrics.plot_roc(y_test, y_prob, figsize=(8, 6), title_fontsize='xx-large', text_fontsize='large', plot_micro=False, plot_macro=False)
plt.legend(['假正例率', '真正例率'], fontsize='large')
plt.show()
```
在上面的代码中,`title_fontsize` 参数设置为 'xx-large',`text_fontsize` 参数设置为 'large',这样就可以将图例的字体大小改为中文。同时,我们在 `plt.legend` 中也添加了一个列表,用于设置图例中的文字,这样就可以将英文的图例改为中文了。
阅读全文