Я знаю, что мой код работает, потому что я попробовал на меньшем наборе данных, однако, когда я пробую его на очень большом наборе данных, получаю следующую ошибку.
code for the random parameter tuning and fitting trained model
#Random search to optimise hyperparameters
estimator = XGBClassifier(
objective= 'binary:logistic',
nthread=4,
)
parameters = {
'max_depth': range (2, 8),
'n_estimators': range(100, 150,5),
'learning_rate': [0.05, 0.1, 0.05]
}
randomized_param = RandomizedSearchCV(estimator=estimator, param_distributions=parameters,
n_iter=5, scoring='neg_mean_squared_error', cv=4, verbose=1)
#Fitting trained model
randomized_param.fit(X, y)
y_predict=randomized_param.predict(X_test)
print(y_predict)
output error
fitFailedWarning: Estimator fit failed. The score on this train-test partition for these parameters will be set to nan. Details:
OSError: [WinError -529697949] Windows Error 0xe06d7363