TypeError：__init__() 为straifiedkfold 的参数“n_splits”获得了多个值

Question

this is my imports

import numpy as np
import matplotlib.pyplot as plt
from matplotlib import patches
from sklearn.mixture import GaussianMixture
from sklearn.model_selection import ShuffleSplit
from sklearn import mixture
from sklearn import datasets
from sklearn.model_selection import KFold
from sklearn.model_selection import StratifiedKFold
from sklearn import mixture




iris = datasets.load_iris()

was folds then change it to splits but error below appear

indices = StratifiedKFold(iris.target, n_splits=5)

train_index, test_index = next(iter(indices))

X_train = iris.data[train_index]
y_train = iris.target[train_index]


X_test = iris.data[test_index]
y_test = iris.target[test_index]

error = TypeError: init () got multiple values for argument 'n_splits'

Answer 1

as per the documnentation , you first need to initialize your StratifiedKFold object with n_splits and then use the skf.get_n_splits method so your code becomes:

skf = StratifiedKFold(n_splits=5)

X = iris.data
y = iris.target 
for train_index, test_index in skf.split(X, y):
    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]