代码之家  ›  专栏  ›  技术社区  ›  Krishi H JakeJ

sklearn中K-折叠交叉验证中各折叠的预测值

  •  2
  • Krishi H JakeJ  · 技术社区  · 7 年前

    我使用python sklearn对一个数据集进行了10次交叉验证,

    result = cross_val_score(best_svr, X, y, cv=10, scoring='r2') print(result.mean())

    我已经得到了r2得分的平均值作为最终结果。我想知道是否有方法打印出每个折叠的预测值(在本例中是10组值)。

    2 回复  |  直到 7 年前
        1
  •  2
  •   jh314    7 年前

    我相信你在找 cross_val_predict 功能。

        2
  •  0
  •   Van Peer    7 年前

    打印每一个折叠的预测,

    for k in range(2,10):
        result = cross_val_score(best_svr, X, y, cv=k, scoring='r2')
        print(k, result.mean())
        y_pred = cross_val_predict(best_svr, X, y, cv=k)
        print(y_pred)