3

我正在尝试使用以下方式绘制平均图：

  1. 计算所有折叠的精确召回曲线。

  2. 计算平均精确召回曲线。我不知道该怎么做，因为不同折叠的尺寸不同。

  3. 绘制曲线，这是在第二步中计算的。

    PS 解决方案在 scikit-learn 中使用交叉验证时绘制 Precision-Recall 曲线不合适，因为如果我计算所有预测的平均值，然后计算 Precision-Recall 曲线，我将得到 AUC = 1.0。这是错误的。

我想得到这样的东西：

from sklearn.metrics import precision_recall_curve
import matplotlib.pyplot as plt

scores = []
for train, test in kfold:
    true, pred = clf.predict(test)
    precision, recall, _ = precision_recall_curve(true, pred)
    scores.append((precision, recall))
precision_avg, recall_avg = compute_average(scores)
plt.plot(precision_avg, recall_avg)
4

0 回答 0