Skip to content

Commit b34c02f

Browse files
author
igor_rukhovich
committed
Added two parameters to xgb benchmarks
1 parent 9d8ca66 commit b34c02f

File tree

7 files changed

+297
-155
lines changed

7 files changed

+297
-155
lines changed

‎configs/cpu_lgbm_gbt_config.json renamed to ‎configs/cpu_lgbm_mb_config.json

Lines changed: 0 additions & 103 deletions
Original file line numberDiff line numberDiff line change
@@ -105,107 +105,4 @@
105105
"objective": ["multiclass"]
106106
}
107107
]
108-
}
109-
110-
111-
{
112-
"algorithm": "xgb_mb",
113-
"dataset": [
114-
{
115-
"source": "csv",
116-
"name": "mortgage1Q",
117-
"training":
118-
{
119-
"x": "../sklbench_data/mortgage_x.csv",
120-
"y": "../sklbench_data/mortgage_y.csv"
121-
}
122-
}
123-
],
124-
"n-estimators": [100],
125-
"objective": ["reg:squarederror"],
126-
"tree-method": ["hist"],
127-
"max-depth": [8],
128-
"scale-pos-weight": [2],
129-
"learning-rate": [0.1],
130-
"subsample": [1],
131-
"reg-alpha": [0.9],
132-
"reg-lambda": [1],
133-
"min-child-weight": [0],
134-
"max-leaves": [256]
135-
},
136-
{
137-
"algorithm": "xgb_mb",
138-
"dataset": [
139-
{
140-
"source": "csv",
141-
"name": "airline-ohe",
142-
"training":
143-
{
144-
"x": "../sklbench_data/airline-ohe_x_train.csv",
145-
"y": "../sklbench_data/airline-ohe_y_train.csv"
146-
}
147-
}
148-
],
149-
"reg-alpha": [0.9],
150-
"max-bin": [256],
151-
"scale-pos-weight": [2],
152-
"learning-rate": [0.1],
153-
"subsample": [1],
154-
"reg-lambda": [1],
155-
"min-child-weight": [0],
156-
"max-depth": [8],
157-
"max-leaves": [256],
158-
"n-estimators": [1000],
159-
"objective": ["binary:logistic"],
160-
"tree-method": ["hist"]
161-
},
162-
{
163-
"algorithm": "xgb_mb",
164-
"dataset": [
165-
{
166-
"source": "csv",
167-
"name": "higgs1m",
168-
"training":
169-
{
170-
"x": "../sklbench_data/higgs1m_x_train.csv",
171-
"y": "../sklbench_data/higgs1m_y_train.csv"
172-
}
173-
}
174-
],
175-
"reg-alpha": [0.9],
176-
"max-bin": [256],
177-
"scale-pos-weight": [2],
178-
"learning-rate": [0.1],
179-
"subsample": [1],
180-
"reg-lambda": [1],
181-
"min-child-weight": [0],
182-
"max-depth": [8],
183-
"max-leaves": [256],
184-
"n-estimators": [1000],
185-
"objective": ["binary:logistic"],
186-
"tree-method": ["hist"]
187-
},
188-
{
189-
"algorithm": "xgb_mb",
190-
"dataset": [
191-
{
192-
"source": "csv",
193-
"name": "msrank",
194-
"training":
195-
{
196-
"x": "../sklbench_data/mlsr_x_train.csv",
197-
"y": "../sklbench_data/mlsr_y_train.csv"
198-
}
199-
}
200-
],
201-
"max-bin": [256],
202-
"learning-rate": [0.3],
203-
"subsample": [1],
204-
"reg-lambda": [2],
205-
"min-child-weight": [1],
206-
"min-split-loss": [0.1],
207-
"max-depth": [8],
208-
"n-estimators": [200],
209-
"objective": ["multi:softprob"],
210-
"tree-method": ["hist"]
211108
}

‎configs/cpu_xgb_config.json

Lines changed: 162 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,162 @@
1+
{
2+
"omp_env": ["OMP_NUM_THREADS", "OMP_PLACES"],
3+
"common": {
4+
"lib": ["xgboost"],
5+
"data-format": ["pandas"],
6+
"data-order": ["F"],
7+
"dtype": ["float32"],
8+
"count-dmatrix": [""]
9+
},
10+
"cases": [
11+
{
12+
"algorithm": "gbt",
13+
"dataset": [
14+
{
15+
"source": "csv",
16+
"name": "plasticc",
17+
"training":
18+
{
19+
"x": "../sklbench_data/plasticc_x_train.csv",
20+
"y": "../sklbench_data/plasticc_y_train.csv"
21+
},
22+
"testing":
23+
{
24+
"x": "../sklbench_data/plasticc_x_test.csv",
25+
"y": "../sklbench_data/plasticc_y_test.csv"
26+
}
27+
}
28+
],
29+
"n-estimators": [60],
30+
"objective": ["multi:softprob"],
31+
"tree-method": ["hist"],
32+
"max-depth": [7],
33+
"subsample": [0.7],
34+
"colsample-bytree": [0.7]
35+
},
36+
{
37+
"algorithm": "gbt",
38+
"dataset": [
39+
{
40+
"source": "csv",
41+
"name": "santander",
42+
"training":
43+
{
44+
"x": "../sklbench_data/santander_x_train.csv",
45+
"y": "../sklbench_data/santander_y_train.csv"
46+
}
47+
}
48+
],
49+
"n-estimators": [10000],
50+
"objective": ["binary:logistic"],
51+
"tree-method": ["hist"],
52+
"max-depth": [1],
53+
"subsample": [0.5],
54+
"eta": [0.1],
55+
"colsample-bytree": [0.05],
56+
"single_precision_histogram": [""]
57+
},
58+
{
59+
"algorithm": "gbt",
60+
"dataset": [
61+
{
62+
"source": "csv",
63+
"name": "mortgage1Q",
64+
"training":
65+
{
66+
"x": "../sklbench_data/mortgage_x.csv",
67+
"y": "../sklbench_data/mortgage_y.csv"
68+
}
69+
}
70+
],
71+
"n-estimators": [100],
72+
"objective": ["reg:squarederror"],
73+
"tree-method": ["hist"],
74+
"max-depth": [8],
75+
"scale-pos-weight": [2],
76+
"learning-rate": [0.1],
77+
"subsample": [1],
78+
"reg-alpha": [0.9],
79+
"reg-lambda": [1],
80+
"min-child-weight": [0],
81+
"max-leaves": [256]
82+
},
83+
{
84+
"algorithm": "gbt",
85+
"dataset": [
86+
{
87+
"source": "csv",
88+
"name": "airline-ohe",
89+
"training":
90+
{
91+
"x": "../sklbench_data/airline-ohe_x_train.csv",
92+
"y": "../sklbench_data/airline-ohe_y_train.csv"
93+
}
94+
}
95+
],
96+
"reg-alpha": [0.9],
97+
"max-bin": [256],
98+
"scale-pos-weight": [2],
99+
"learning-rate": [0.1],
100+
"subsample": [1],
101+
"reg-lambda": [1],
102+
"min-child-weight": [0],
103+
"max-depth": [8],
104+
"max-leaves": [256],
105+
"n-estimators": [1000],
106+
"objective": ["binary:logistic"],
107+
"tree-method": ["hist"]
108+
},
109+
{
110+
"algorithm": "gbt",
111+
"dataset": [
112+
{
113+
"source": "csv",
114+
"name": "higgs1m",
115+
"training":
116+
{
117+
"x": "../sklbench_data/higgs1m_x_train.csv",
118+
"y": "../sklbench_data/higgs1m_y_train.csv"
119+
}
120+
}
121+
],
122+
"reg-alpha": [0.9],
123+
"max-bin": [256],
124+
"scale-pos-weight": [2],
125+
"learning-rate": [0.1],
126+
"subsample": [1],
127+
"reg-lambda": [1],
128+
"min-child-weight": [0],
129+
"max-depth": [8],
130+
"max-leaves": [256],
131+
"n-estimators": [1000],
132+
"objective": ["binary:logistic"],
133+
"tree-method": ["hist"],
134+
"enable_experimental_json_serialization": ["False"]
135+
},
136+
{
137+
"algorithm": "gbt",
138+
"dataset": [
139+
{
140+
"source": "csv",
141+
"name": "msrank",
142+
"training":
143+
{
144+
"x": "../sklbench_data/mlsr_x_train.csv",
145+
"y": "../sklbench_data/mlsr_y_train.csv"
146+
}
147+
}
148+
],
149+
"max-bin": [256],
150+
"learning-rate": [0.3],
151+
"subsample": [1],
152+
"reg-lambda": [2],
153+
"min-child-weight": [1],
154+
"min-split-loss": [0.1],
155+
"max-depth": [8],
156+
"n-estimators": [200],
157+
"objective": ["multi:softprob"],
158+
"tree-method": ["hist"],
159+
"single_precision_histogram": [""]
160+
}
161+
]
162+
}

‎configs/cpu_xgb_gbt_config.json

Lines changed: 0 additions & 38 deletions
This file was deleted.

0 commit comments

Comments
 (0)