ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-02-07 13:23:40 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update XGBoost example (#524)

Browse Source 
* Change dataset (load_boston is deprecated)
* Black format
* Add save_model
This commit is contained in:
pollfly 2021-12-23 14:07:42 +02:00 committed by GitHub
parent 8f1e2b0aae
commit b036bbce2f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 14 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
examples/frameworks/xgboost/xgboost_metrics.py
View File

@ -1,24 +1,27 @@
import xgboost as xgb
from sklearn.datasets import load_boston
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
from clearml import Task
task = Task.init(project_name='examples', task_name='xgboost metric auto reporting')
task = Task.init(project_name="examples", task_name="xgboost metric auto reporting")
X, y = load_boston(return_X_y=True)
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=100)
X, y = load_iris(return_X_y=True)
X_train, X_test, y_train, y_test = train_test_split(
X, y, test_size=0.2, random_state=100
)
dtrain = xgb.DMatrix(X_train, label=y_train)
dtest = xgb.DMatrix(X_test, label=y_test)
params = {
'objective': 'reg:squarederror',
'eval_metric': 'rmse'
}
params = {"objective": "reg:squarederror", "eval_metric": "rmse"}
bst = xgb.train(
params, dtrain, num_boost_round=100,
evals=[(dtrain, 'train'), (dtest, 'test')],
verbose_eval=0
params,
dtrain,
num_boost_round=100,
evals=[(dtrain, "train"), (dtest, "test")],
verbose_eval=0,
)
bst.save_model("best_model")