Skip to content

Commit dc7504c

Browse files
AlexsandrussAlexsandruss
committed
modelbuilders configs
1 parent 363b6b0 commit dc7504c

File tree

3 files changed

+229
-3
lines changed

3 files changed

+229
-3
lines changed

‎configs/lgbm_mb_cpu_config.json

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

‎configs/xgb_cpu_config.json

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,8 @@
5252
"max-depth": [1],
5353
"subsample": [0.5],
5454
"eta": [0.1],
55-
"colsample-bytree": [0.05]
55+
"colsample-bytree": [0.05],
56+
"single-precision-histogram": [""]
5657
},
5758
{
5859
"algorithm": "gbt",
@@ -129,7 +130,8 @@
129130
"max-leaves": [256],
130131
"n-estimators": [1000],
131132
"objective": ["binary:logistic"],
132-
"tree-method": ["hist"]
133+
"tree-method": ["hist"],
134+
"enable-experimental-json-serialization": ["False"]
133135
},
134136
{
135137
"algorithm": "gbt",
@@ -153,7 +155,8 @@
153155
"max-depth": [8],
154156
"n-estimators": [200],
155157
"objective": ["multi:softprob"],
156-
"tree-method": ["hist"]
158+
"tree-method": ["hist"],
159+
"single-precision-histogram": [""]
157160
}
158161
]
159162
}

‎configs/xgb_mb_cpu_config.json

Lines changed: 114 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,114 @@
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": "data/mortgage_x.csv",
20+
"y": "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": "data/airline-ohe_x_train.csv",
45+
"y": "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": "data/higgs1m_x_train.csv",
71+
"y": "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+
"enable-experimental-json-serialization": ["False"]
88+
},
89+
{
90+
"algorithm": "xgb_mb",
91+
"dataset": [
92+
{
93+
"source": "csv",
94+
"name": "msrank",
95+
"training":
96+
{
97+
"x": "data/mlsr_x_train.csv",
98+
"y": "data/mlsr_y_train.csv"
99+
}
100+
}
101+
],
102+
"max-bin": [256],
103+
"learning-rate": [0.3],
104+
"subsample": [1],
105+
"reg-lambda": [2],
106+
"min-child-weight": [1],
107+
"min-split-loss": [0.1],
108+
"max-depth": [8],
109+
"n-estimators": [200],
110+
"objective": ["multi:softprob"],
111+
"tree-method": ["hist"]
112+
}
113+
]
114+
}

0 commit comments

Comments
 (0)