в DS
import pandas as pd
from sklearn.ensemble import RandomForestClassifier
from sklearn.model_selection import GridSearchCV
df = pd.read_csv(r'C:/Users/CptCu/Desktop/training_mush.csv')
X_train = df.drop('class', axis=1)
y_train = df['class']
clf = RandomForestClassifier()
parametrs = { 'n_estimators': range (10, 51, 10),
'max_depth': range (1,13, 2),
'min_samples_leaf': range (1,8),
'min_samples_split': range (2,10,2) }
grid = GridSearchCV(clf, parametrs, cv=5)
grid.fit(X_train, y_train)grid.best_params_
{'criterion': 'entropy',
'max_depth': 5,
'min_samples_leaf': 3,
'min_samples_split': 2,
'n_estimators': 30}
В строке
grid.fit(X_train,y_train)grid.best_param_
Ругается на синтаксис. Можете подсказать в каком направлении копать?
P.s. Если знаете какой-нибудь чатик для начинающих - буду признателен
Перенесите grid.best… на отдельную строчку.
Обсуждают сегодня