← На главную Воспоминания

Вторник, 10 Октября 2023

Закончил изучать XGBoost. Начал изучать утечку данных

from xgboost import XGBRegressor
my_model = XGBRegressor(
    n_estimators=1000, 
    learning_rate=0.005, 
    n_jobs=4
)
my_model.fit(X_train, y_train, 
    early_stopping_rounds=5, 
    eval_set=[(X_valid, y_valid)], 
    verbose=False
)