TypeError: объект 'rv_frozen' нельзя интерпретировать как целое число - PullRequest
0 голосов
/ 07 мая 2020

Я попытался запустить свой Grid Search CV с установленным параметром. Но почему-то он говорит TypeError: 'rv_frozen' object cannot be interpreted as an integer, что происходит в строке grid.fit(X_train,y_train). Почему примерка возвращает ошибку? Что мне делать, чтобы это исправить?

final['cluster'] = dbcpca
final = pd.get_dummies(final)
# sns.heatmap(final.corr(),annot=True)


X = np.array(final.drop(['unemployed'],1))
y = np.array(final['unemployed'])

X = StandardScaler().fit_transform(X)
y = StandardScaler().fit_transform(y.reshape(-1,1))

X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2)

xgb_model = xgb.XGBRegressor()

params = {
    "colsample_bytree": [uniform(0.7, 0.3)],
    "gamma": [uniform(0, 0.5)],
    "learning_rate": [uniform(0.03, 0.3)], # default 0.1 
    "max_depth": [randint(2, 6)], # default 3
    "n_estimators": [randint(100, 150)], # default 100
    "subsample": [uniform(0.6, 0.4)]
}

grid = GridSearchCV(estimator= xgb_model,param_grid =params,
                    refit=True,verbose=1,cv =10,n_jobs=-1)
grid.fit(X_train,y_train)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...