text_plain_output_1.png
import xgboost as xgb

model = xgb.XGBRegressor(objective='reg:tweedie', tree_method='gpu_hist', predictor='gpu_predictor', sampling_method='gradient_based', max_bin=512, silent=False, random_state=17)
model.fit(X_train, y_train)
preds_test = model.predict(X_test)
preds_train = model.predict(X_train)