from sklearn.model_selection import KFold
import numpy as np
# Create some dummy data
X = np.array([[1, 2], [3, 4], [1, 2], [3, 4], [0, 0])
# Remove rows having all of their columns equal to 0
X = X[~np.all(X == 0, axis=1)]
# Assuming 2-fold cross-validation
kf = KFold(n_splits=2)
kf.get_n_splits(X)
现在
^{pr2}$kf
有两个训练/测试折叠:相关问题 更多 >
编程相关推荐