Skip to content

Commit 363b6b0

Browse files
committed
Correct knn configs
1 parent 27c9efb commit 363b6b0

File tree

2 files changed

+43
-1
lines changed

2 files changed

+43
-1
lines changed

configs/cuml_config.json

Lines changed: 22 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -350,6 +350,9 @@
350350
"n_features": 3,
351351
"training": {
352352
"n_samples": 100000
353+
},
354+
"testing": {
355+
"n_samples": 100000
353356
}
354357
},
355358
{
@@ -359,6 +362,9 @@
359362
"n_features": 10,
360363
"training": {
361364
"n_samples": 100000
365+
},
366+
"testing": {
367+
"n_samples": 100000
362368
}
363369
},
364370
{
@@ -368,10 +374,25 @@
368374
"n_features": 50,
369375
"training": {
370376
"n_samples": 20000
377+
},
378+
"testing": {
379+
"n_samples": 20000
380+
}
381+
},
382+
{
383+
"source": "synthetic",
384+
"type": "classification",
385+
"n_classes": 10,
386+
"n_features": 16,
387+
"training": {
388+
"n_samples": 250000
389+
},
390+
"testing": {
391+
"n_samples": 250000
371392
}
372393
}
373394
],
374-
"method": ["brute", "kd_tree"]
395+
"method": ["brute"]
375396
},
376397
{
377398
"algorithm": "train_test_split",

configs/skl_config.json

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -418,6 +418,9 @@
418418
"n_features": 3,
419419
"training": {
420420
"n_samples": 100000
421+
},
422+
"testing": {
423+
"n_samples": 100000
421424
}
422425
},
423426
{
@@ -427,6 +430,9 @@
427430
"n_features": 10,
428431
"training": {
429432
"n_samples": 100000
433+
},
434+
"testing": {
435+
"n_samples": 100000
430436
}
431437
},
432438
{
@@ -436,6 +442,21 @@
436442
"n_features": 50,
437443
"training": {
438444
"n_samples": 20000
445+
},
446+
"testing": {
447+
"n_samples": 20000
448+
}
449+
},
450+
{
451+
"source": "synthetic",
452+
"type": "classification",
453+
"n_classes": 10,
454+
"n_features": 16,
455+
"training": {
456+
"n_samples": 250000
457+
},
458+
"testing": {
459+
"n_samples": 250000
439460
}
440461
}
441462
],

0 commit comments

Comments
 (0)