分层K折后该怎么办?

时间:2020-04-02 08:27:03

标签: machine-learning scikit-learn cross-validation k-fold

我已使用StratifiedKFold对我的训练数据集进行交叉验证。该模型已达到75%的准确度,我发现这是可以接受的。

我应该继续并将模型实现到测试集上吗?

model.fit(X_train, y_train)

y_pred = model.predict(X_test)

accuracy = accuracy_score(y_test, y_pred)

还是我应该做其他事情来执行分层k折以获得新的测试集?

0 个答案:

没有答案