ax.set_ylabel('Predicted')
plt.show()
KFlod的例子
Stratified k-fold:实现了分层交叉切分
from sklearn.model_selection import StratifiedKFold...54],
[61, 62, 63, 64],
[71, 72, 73, 74]])
y = np.array([1, 1, 0, 0, 1, 1, 0, 0])
stratified_folder = StratifiedKFold...7]
Stratified y_train: [1 1 0 0 1 0]
Stratified y_test: [1 0]
from sklearn.model_selection import StratifiedKFold...54],
[61, 62, 63, 64],
[71, 72, 73, 74]])
y = np.array([1, 1, 0, 0, 1, 1, 0, 0])
stratified_folder = StratifiedKFold