我使用 Stratified K fold 来学习模型。下面是python代码:
>def stratified_cv_v1(X, y, clf, shuffle=True, n=10,):
> stratified_k_fold = StratifiedKFold(n_splits=n,shuffle=shuffle)
> y_pred_v1 = y.copy()
> for ii, jj in stratified_k_fold.split(X,y):
> X_train, X_test = X[ii], X[jj]
> y_train = y[ii]
> clf_v2 = clf()
> clf_v2.fit(X_train,y_train)
> y_pred[jj] = clf.predict(X_test)
> return y_pred_v1
>print(classification_report(y, stratified_cv_v1(X, y, GradientBoostingClassifier)))
现在如何使用该模型在需要预测的新数据集上进行部署?