推荐：一个画出高大上的机器学习图的神器（论文必备）

本人在写论文的时候，很多图片是用matplotlib和seaborn画的，但是，我还有一个神器，Scikit-plot，通过这个神器，画出了更加高大上的机器学习图，本文对Scikit-plot做下简单介绍。（作者：黄海广）

安装说明

安装Scikit-plot非常简单，直接用命令：

pip install scikit-plot

即可完成安装。

仓库地址：

https://github.com/reiinakano/scikit-plot

里面有使用说明和样例（py和ipynb格式）。

使用说明

简单举几个例子

from sklearn.datasets import load_digitsfrom sklearn.model_selection import train_test_splitfrom sklearn.naive_bayes import GaussianNBX, y = load_digits(return_X_y=True)X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.33)nb = GaussianNB()nb.fit(X_train, y_train)predicted_probas = nb.predict_proba(X_test)# The magic happens hereimport matplotlib.pyplot as pltimport scikitplot as skpltskplt.metrics.plot_roc(y_test, predicted_probas)plt.show()import load_digits
from sklearn.model_selection import train_test_split
from sklearn.naive_bayes import GaussianNB
X, y = load_digits(return_X_y=True)
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.33)
nb = GaussianNB()
nb.fit(X_train, y_train)
predicted_probas = nb.predict_proba(X_test)
# The magic happens here
import matplotlib.pyplot as plt
import scikitplot as skplt
skplt.metrics.plot_roc(y_test, predicted_probas)
plt.show()

效果如图（相当高大上！）

图：ROC曲线

import matplotlib.pyplot as pltfrom sklearn.naive_bayes import GaussianNBfrom sklearn.datasets import load_digits as load_dataimport scikitplot as skplt# Load datasetX, y = load_data(return_X_y=True)# Create classifier instance then fitnb = GaussianNB()nb.fit(X,y)# Get predicted probabilitiesy_probas = nb.predict_proba(X)skplt.metrics.plot_precision_recall_curve(y, y_probas, cmap='nipy_spectral')plt.show()as plt
from sklearn.naive_bayes import GaussianNB
from sklearn.datasets import load_digits as load_data
import scikitplot as skplt
# Load dataset
X, y = load_data(return_X_y=True)
# Create classifier instance then fit
nb = GaussianNB()
nb.fit(X,y)
# Get predicted probabilities
y_probas = nb.predict_proba(X)
skplt.metrics.plot_precision_recall_curve(y, y_probas, cmap='nipy_spectral')
plt.show()

图：P-R曲线

from sklearn.ensemble import RandomForestClassifierfrom sklearn.datasets import load_digits as load_datafrom sklearn.model_selection import cross_val_predictimport matplotlib.pyplot as pltimport scikitplot as skpltX, y = load_data(return_X_y=True)# Create an instance of the RandomForestClassifierclassifier = RandomForestClassifier()# Perform predictionspredictions = cross_val_predict(classifier, X, y)plot = skplt.metrics.plot_confusion_matrix(y, predictions, normalize=True)plt.show()import RandomForestClassifier
from sklearn.datasets import load_digits as load_data
from sklearn.model_selection import cross_val_predict
import matplotlib.pyplot as plt
import scikitplot as skplt
X, y = load_data(return_X_y=True)
# Create an instance of the RandomForestClassifier
classifier = RandomForestClassifier()
# Perform predictions
predictions = cross_val_predict(classifier, X, y)
plot = skplt.metrics.plot_confusion_matrix(y, predictions, normalize=True)
plt.show()

图：归一化混淆矩阵

图：学习曲线、特征重要性

图：K-means肘点图

总结

本文对Scikit-plot做下简单介绍，这是一个机器学习的画图神器，几行代码就能画出高大上的机器学习图，作者当年的博士论文也是靠这个画图的。

仓库地址：

https://github.com/reiinakano/scikit-plot

里面有使用说明和样例。