image_output_1.png
import xgboost

import xgboost
xgBoost = xgboost.XGBRegressor(max_depth=3, learning_rate=0.1, n_estimators=100, booster='gbtree')
xgBoost.fit(X_train, Y_train)
print('train score', xgBoost.score(X_train, Y_train))
print('test score', xgBoost.score(X_test, Y_test))