transform () принимает 2 позиционных аргумента, но 3 было дано

Question

Ниже мой код: -

from sklearn.feature_selection import SelectKBest, chi2, f_regression

X_train_new = SelectKBest(score_func=chi2,k=2000).fit_transform(X_train_2,y_train)

X_cv_new = SelectKBest(score_func=chi2,k=2000).transform(X_cv_2,y_cv)

X_test_new = SelectKBest(score_func=chi2,k=2000).transform(X_test_2,y_test)

X_train_new.shape, X_cv_new.shape, X_test_new.shape

Я пытаюсь выбрать лучшие 2000 функций, а затем применить их для обученных поездом tfidf данных, резюме и данных испытаний.

При выполнении вышеупомянутого я получаю "TypeError: transform() takes 2 positional arguments but 3 were given"

Answer 1

Исходя из документации, self считается аргументом, поэтому, когда вы предоставляете y_train, y_cv, и y_test в метод, эти принимаются в качестве третьего аргумента. Метод принимает ( self , X), где X - массив формы [n_samples, n_features].