sklearn KFold

未空blog

已于 2022-03-07 09:23:19 修改

阅读量532

点赞数

分类专栏：机器学习文章标签： sklearn 人工智能 python

于 2022-03-07 09:02:16 首次发布

本文链接：https://blog.csdn.net/weixin_42324904/article/details/123322095

版权

机器学习专栏收录该内容

2 篇文章 0 订阅

订阅专栏

from sklearn.model_selection import KFold
import numpy as np
X = np.array([[1, 2], [3, 4], [5, 6], [7, 8], [9, 10],[11, 12]])
y = np.array([1, 2, 3, 4, 5, 6])
kf = KFold(n_splits=3)
print( kf.get_n_splits(X))
kfs = kf.split(X)

for train_index, test_index in kf.split(X):
	print("TRAIN:", train_index, "TEST:", test_index)
	X_train, X_test = X[train_index], X[test_index]
	y_train, y_test = y[train_index], y[test_index]

print()

一目了然

不言自喻

不了然就debug一下

还有个小问题

random_state参数的神奇: 随机切割、KFold和机器学习_cs12ab的博客-CSDN博客

from sklearn.model_selection import KFold
import numpy as np
X = np.array([[1, 2], [3, 4], [5, 6], [7, 8], [9, 10],[11, 12], [13,14], [15, 16]])
y = np.array([1, 2, 3, 4, 5, 6, 7, 8])
kf = KFold(n_splits=8, shuffle=True, random_state=3)
print( kf.get_n_splits(X))
kfs = kf.split(X)

for train_index, test_index in kf.split(X):
	print("TRAIN:", train_index, "TEST:", test_index)
	X_train, X_test = X[train_index], X[test_index]
	y_train, y_test = y[train_index], y[test_index]

print()

random state是种子，不会产生影响