def _get_split(X, y):
split = ShuffleSplit(y.shape[0], n_iter=1)
train, validate = list(split)[0]
X_train, X_validate, y_train, y_validate = X[train], X[validate], y[train], y[validate]
return X_train, X_validate, y_train, y_validate
评论列表
文章目录