Блог

Главная - Вопросы по программированию - Ошибка типа: __init__() получил неожиданный аргумент ключевого слова ‘average’

Ошибка типа: __init__() получил неожиданный аргумент ключевого слова ‘average’

#python-3.x #machine-learning #svm #random-forest

#python-3.x #машинное обучение #svm #случайный лес

Вопрос:

При выполнении следующего кода я получаю сообщение об ошибке TypeError: __init__() got an unexpected keyword argument 'average'

Что я делаю не так?

 from sklearn.model_selection import GridSearchCV

X_train, X_holdout, y_train, y_holdout = train_test_split(X, y, test_size=0.3, random_state=17)

tree = DecisionTreeClassifier(random_state=17)

tree_params = {'max_depth': range(1,5), 'max_features': range(3,6), 'criterion': ['gini','entropy']}

tree_grid = GridSearchCV(tree, tree_params, cv=scoring='recall',average='macro')

tree_grid.fit(X_train, y_train)

Ответ №1:

У GridSearchCV класса нет average параметра — вы используете его в average='macro' .

Пожалуйста, взгляните на документацию класса:

 class sklearn.model_selection.GridSearchCV(estimator, param_grid, *, scoring=None, n_jobs=None, iid='deprecated', refit=True, cv=None, verbose=0, pre_dispatch='2*n_jobs', error_score=nan, return_train_score=False)
Метки: