Skip to content

Commit d0b6c40

Browse files
author
igor_rukhovich
committed
Added modelbuilders benchmarks for xgb and lgbm
1 parent 1423837 commit d0b6c40

File tree

7 files changed

+1045
-1
lines changed

7 files changed

+1045
-1
lines changed

‎.gitignore

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,3 +11,4 @@ __work*
1111
# Datasets
1212
dataset
1313
*.csv
14+
*.npy

‎LICENSE

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
MIT License
22

3-
Copyright (c) 2017-2019 Intel Corporation
3+
Copyright (c) 2017-2020 Intel Corporation
44

55
Permission is hereby granted, free of charge, to any person obtaining a copy
66
of this software and associated documentation files (the "Software"), to deal

‎configs/cpu_lgbm_gbt_config.json

Lines changed: 113 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,113 @@
1+
{
2+
"omp_env": ["OMP_NUM_THREADS", "OMP_PLACES"],
3+
"common": {
4+
"lib": ["modelbuilders"],
5+
"data-format": ["pandas"],
6+
"data-order": ["F"],
7+
"dtype": ["float32"],
8+
"count-dmatrix": [""]
9+
},
10+
"cases": [
11+
{
12+
"algorithm": "lgbm_mb",
13+
"dataset": [
14+
{
15+
"source": "csv",
16+
"name": "mortgage1Q",
17+
"training":
18+
{
19+
"x": "../sklbench_data/mortgage_x.csv",
20+
"y": "../sklbench_data/mortgage_y.csv"
21+
}
22+
}
23+
],
24+
"n-estimators": [100],
25+
"objective": ["reg:squarederror"],
26+
"tree-method": ["hist"],
27+
"max-depth": [8],
28+
"scale-pos-weight": [2],
29+
"learning-rate": [0.1],
30+
"subsample": [1],
31+
"reg-alpha": [0.9],
32+
"reg-lambda": [1],
33+
"min-child-weight": [0],
34+
"max-leaves": [256]
35+
},
36+
{
37+
"algorithm": "lgbm_mb",
38+
"dataset": [
39+
{
40+
"source": "csv",
41+
"name": "airline-ohe",
42+
"training":
43+
{
44+
"x": "../sklbench_data/airline-ohe_x_train.csv",
45+
"y": "../sklbench_data/airline-ohe_y_train.csv"
46+
}
47+
}
48+
],
49+
"reg-alpha": [0.9],
50+
"max-bin": [256],
51+
"scale-pos-weight": [2],
52+
"learning-rate": [0.1],
53+
"subsample": [1],
54+
"reg-lambda": [1],
55+
"min-child-weight": [0],
56+
"max-depth": [8],
57+
"max-leaves": [256],
58+
"n-estimators": [1000],
59+
"objective": ["binary:logistic"],
60+
"tree-method": ["hist"]
61+
},
62+
{
63+
"algorithm": "lgbm_mb",
64+
"dataset": [
65+
{
66+
"source": "csv",
67+
"name": "higgs1m",
68+
"training":
69+
{
70+
"x": "../sklbench_data/higgs1m_x_train.csv",
71+
"y": "../sklbench_data/higgs1m_y_train.csv"
72+
}
73+
}
74+
],
75+
"reg-alpha": [0.9],
76+
"max-bin": [256],
77+
"scale-pos-weight": [2],
78+
"learning-rate": [0.1],
79+
"subsample": [1],
80+
"reg-lambda": [1],
81+
"min-child-weight": [0],
82+
"max-depth": [8],
83+
"max-leaves": [256],
84+
"n-estimators": [1000],
85+
"objective": ["binary:logistic"],
86+
"tree-method": ["hist"]
87+
},
88+
{
89+
"algorithm": "lgbm_mb",
90+
"dataset": [
91+
{
92+
"source": "csv",
93+
"name": "msrank",
94+
"training":
95+
{
96+
"x": "../sklbench_data/mlsr_x_train.csv",
97+
"y": "../sklbench_data/mlsr_y_train.csv"
98+
}
99+
}
100+
],
101+
"max-bin": [256],
102+
"learning-rate": [0.3],
103+
"subsample": [1],
104+
"reg-lambda": [2],
105+
"min-child-weight": [1],
106+
"min-split-loss": [0.1],
107+
"max-depth": [8],
108+
"n-estimators": [200],
109+
"objective": ["multi:softprob"],
110+
"tree-method": ["hist"]
111+
}
112+
]
113+
}

‎configs/cpu_xgb_gbt_config.json

Lines changed: 113 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,113 @@
1+
{
2+
"omp_env": ["OMP_NUM_THREADS", "OMP_PLACES"],
3+
"common": {
4+
"lib": ["modelbuilders"],
5+
"data-format": ["pandas"],
6+
"data-order": ["F"],
7+
"dtype": ["float32"],
8+
"count-dmatrix": [""]
9+
},
10+
"cases": [
11+
{
12+
"algorithm": "xgb_mb",
13+
"dataset": [
14+
{
15+
"source": "csv",
16+
"name": "mortgage1Q",
17+
"training":
18+
{
19+
"x": "../sklbench_data/mortgage_x.csv",
20+
"y": "../sklbench_data/mortgage_y.csv"
21+
}
22+
}
23+
],
24+
"n-estimators": [100],
25+
"objective": ["reg:squarederror"],
26+
"tree-method": ["hist"],
27+
"max-depth": [8],
28+
"scale-pos-weight": [2],
29+
"learning-rate": [0.1],
30+
"subsample": [1],
31+
"reg-alpha": [0.9],
32+
"reg-lambda": [1],
33+
"min-child-weight": [0],
34+
"max-leaves": [256]
35+
},
36+
{
37+
"algorithm": "xgb_mb",
38+
"dataset": [
39+
{
40+
"source": "csv",
41+
"name": "airline-ohe",
42+
"training":
43+
{
44+
"x": "../sklbench_data/airline-ohe_x_train.csv",
45+
"y": "../sklbench_data/airline-ohe_y_train.csv"
46+
}
47+
}
48+
],
49+
"reg-alpha": [0.9],
50+
"max-bin": [256],
51+
"scale-pos-weight": [2],
52+
"learning-rate": [0.1],
53+
"subsample": [1],
54+
"reg-lambda": [1],
55+
"min-child-weight": [0],
56+
"max-depth": [8],
57+
"max-leaves": [256],
58+
"n-estimators": [1000],
59+
"objective": ["binary:logistic"],
60+
"tree-method": ["hist"]
61+
},
62+
{
63+
"algorithm": "xgb_mb",
64+
"dataset": [
65+
{
66+
"source": "csv",
67+
"name": "higgs1m",
68+
"training":
69+
{
70+
"x": "../sklbench_data/higgs1m_x_train.csv",
71+
"y": "../sklbench_data/higgs1m_y_train.csv"
72+
}
73+
}
74+
],
75+
"reg-alpha": [0.9],
76+
"max-bin": [256],
77+
"scale-pos-weight": [2],
78+
"learning-rate": [0.1],
79+
"subsample": [1],
80+
"reg-lambda": [1],
81+
"min-child-weight": [0],
82+
"max-depth": [8],
83+
"max-leaves": [256],
84+
"n-estimators": [1000],
85+
"objective": ["binary:logistic"],
86+
"tree-method": ["hist"]
87+
},
88+
{
89+
"algorithm": "xgb_mb",
90+
"dataset": [
91+
{
92+
"source": "csv",
93+
"name": "msrank",
94+
"training":
95+
{
96+
"x": "../sklbench_data/mlsr_x_train.csv",
97+
"y": "../sklbench_data/mlsr_y_train.csv"
98+
}
99+
}
100+
],
101+
"max-bin": [256],
102+
"learning-rate": [0.3],
103+
"subsample": [1],
104+
"reg-lambda": [2],
105+
"min-child-weight": [1],
106+
"min-split-loss": [0.1],
107+
"max-depth": [8],
108+
"n-estimators": [200],
109+
"objective": ["multi:softprob"],
110+
"tree-method": ["hist"]
111+
}
112+
]
113+
}

0 commit comments

Comments
 (0)