From 9b1190b39775c1e72f3eaa7bf70f3aa642681d41 Mon Sep 17 00:00:00 2001 From: Igor Rukhovich Date: Fri, 16 Jul 2021 13:52:58 +0300 Subject: [PATCH] Additional gpu config + codeowner --- .github/CODEOWNERS | 7 +- .../xgboost/xgb_gpu_additional_config.json | 141 ++++++++++++++++++ ...u_config.json => xgb_gpu_main_config.json} | 0 3 files changed, 145 insertions(+), 3 deletions(-) create mode 100644 configs/xgboost/xgb_gpu_additional_config.json rename configs/xgboost/{xgb_gpu_config.json => xgb_gpu_main_config.json} (100%) diff --git a/.github/CODEOWNERS b/.github/CODEOWNERS index 914918261..aacf53040 100755 --- a/.github/CODEOWNERS +++ b/.github/CODEOWNERS @@ -1,8 +1,9 @@ #owners and reviewers -sklearn_bench/* @PetrovKP @Alexsandruss -daal4py_bench/* @PetrovKP @Alexsandruss cuml_bench/* @PetrovKP @Alexsandruss -datasets/* @PetrovKP @Alexsandruss +daal4py_bench/* @PetrovKP @Alexsandruss +datasets/* @PetrovKP @Alexsandruss @RukhovichIV modelbuilders_bench/* @ShvetsKS @RukhovichIV +report_generator/* @PetrovKP @Alexsandruss @RukhovichIV +sklearn_bench/* @PetrovKP @Alexsandruss xgboost_bench/* @ShvetsKS @RukhovichIV *.md @outoftardis diff --git a/configs/xgboost/xgb_gpu_additional_config.json b/configs/xgboost/xgb_gpu_additional_config.json new file mode 100644 index 000000000..75036ad4b --- /dev/null +++ b/configs/xgboost/xgb_gpu_additional_config.json @@ -0,0 +1,141 @@ +{ + "common": { + "lib": "xgboost", + "data-format": "cudf", + "data-order": "F", + "dtype": "float32", + "algorithm": "gbt", + "tree-method": "gpu_hist", + "count-dmatrix": "", + "max-depth": 8, + "learning-rate": 0.1, + "reg-lambda": 1, + "max-leaves": 256 + }, + "cases": [ + { + "objective": "binary:logistic", + "scale-pos-weight": 2.1067817411664587, + "dataset": [ + { + "source": "npy", + "name": "airline", + "training": { + "x": "data/airline_x_train.npy", + "y": "data/airline_y_train.npy" + }, + "testing": { + "x": "data/airline_x_test.npy", + "y": "data/airline_y_test.npy" + } + } + ] + }, + { + "objective": "binary:logistic", + "scale-pos-weight": 173.63348001466812, + "dataset": [ + { + "source": "npy", + "name": "bosch", + "training": { + "x": "data/bosch_x_train.npy", + "y": "data/bosch_y_train.npy" + }, + "testing": { + "x": "data/bosch_x_test.npy", + "y": "data/bosch_y_test.npy" + } + } + ] + }, + { + "objective": "multi:softmax", + "dataset": [ + { + "source": "npy", + "name": "covtype", + "training": { + "x": "data/covtype_x_train.npy", + "y": "data/covtype_y_train.npy" + }, + "testing": { + "x": "data/covtype_x_test.npy", + "y": "data/covtype_y_test.npy" + } + } + ] + }, + { + "objective": "binary:logistic", + "scale-pos-weight": 2.0017715678375363, + "dataset": [ + { + "source": "npy", + "name": "epsilon", + "training": { + "x": "data/epsilon_x_train.npy", + "y": "data/epsilon_y_train.npy" + }, + "testing": { + "x": "data/epsilon_x_test.npy", + "y": "data/epsilon_y_test.npy" + } + } + ] + }, + { + "objective": "binary:logistic", + "scale-pos-weight": 578.2868020304569, + "dataset": [ + { + "source": "npy", + "name": "fraud", + "training": { + "x": "data/fraud_x_train.npy", + "y": "data/fraud_y_train.npy" + }, + "testing": { + "x": "data/fraud_x_test.npy", + "y": "data/fraud_y_test.npy" + } + } + ] + }, + { + "objective": "binary:logistic", + "scale-pos-weight": 1.8872389605086624, + "dataset": [ + { + "source": "npy", + "name": "higgs", + "training": { + "x": "data/higgs_x_train.npy", + "y": "data/higgs_y_train.npy" + }, + "testing": { + "x": "data/higgs_x_test.npy", + "y": "data/higgs_y_test.npy" + } + } + ] + }, + { + "objective": "reg:squarederror", + "dataset": [ + { + "source": "npy", + "name": "year_prediction_msd", + "training": { + "x": "data/year_prediction_msd_x_train.npy", + "y": "data/year_prediction_msd_y_train.npy" + }, + "testing": { + "x": "data/year_prediction_msd_x_test.npy", + "y": "data/year_prediction_msd_y_test.npy" + } + } + ] + } + ] +} diff --git a/configs/xgboost/xgb_gpu_config.json b/configs/xgboost/xgb_gpu_main_config.json similarity index 100% rename from configs/xgboost/xgb_gpu_config.json rename to configs/xgboost/xgb_gpu_main_config.json