Skip to content

Commit 301e918

Browse files
committed
Add train_test_split in config
1 parent 294b6ee commit 301e918

File tree

1 file changed

+37
-0
lines changed

1 file changed

+37
-0
lines changed

config_example.json

100755100644
Lines changed: 37 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -176,6 +176,43 @@
176176
"max-cache-size": [4],
177177
"kernel": ["rbf"]
178178
},
179+
{
180+
"lib": ["sklearn"],
181+
"algorithm": "train_test_split",
182+
"dataset": [
183+
{
184+
"source": "synthetic",
185+
"type": "classification",
186+
"n_classes": 2,
187+
"n_features": 20,
188+
"training": {
189+
"n_samples": 500000
190+
}
191+
},
192+
{
193+
"source": "synthetic",
194+
"type": "classification",
195+
"n_classes": 2,
196+
"n_features": 100,
197+
"training": {
198+
"n_samples": 100000
199+
}
200+
},
201+
{
202+
"source": "synthetic",
203+
"type": "classification",
204+
"n_classes": 2,
205+
"n_features": 5000,
206+
"training": {
207+
"n_samples": 5000
208+
}
209+
}
210+
],
211+
"shuffle": [""],
212+
"include-y": [""],
213+
"train-size": [0.75],
214+
"test-size": [0.25]
215+
},
179216
{
180217
"lib": ["xgboost"],
181218
"algorithm": "gbt",

0 commit comments

Comments
 (0)