Doctorado-ML/benchmark
1
0
Fork 0
mirror of https://github.com/Doctorado-ML/benchmark.git synced 2025-08-17 08:25:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add njobs and nestimators to build_grid

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2022-03-10 12:07:09 +01:00
parent 3e72ba613d
commit 32a82ceb3c
2 changed files with 25 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
results/grid_input_accuracy_ODTE.json
View File

@@ -1,5 +1,11 @@
[
{
"n_jobs": [
-1
],
"n_estimators": [
100
],
"base_estimator__C": [
1.0
],
@@ -11,6 +17,12 @@
]
},
{
"n_jobs": [
-1
],
"n_estimators": [
100
],
"base_estimator__C": [
0.001,
0.0275,
@@ -32,6 +44,12 @@
]
},
{
"n_jobs": [
-1
],
"n_estimators": [
100
],
"base_estimator__C": [
0.05,
1.0,
@@ -60,6 +78,12 @@
]
},
{
"n_jobs": [
-1
],
"n_estimators": [
100
],
"base_estimator__C": [
0.05,
0.2,

2
src/build_grid.py
View File

@@ -81,7 +81,7 @@ for kernel in kernels:
results[kernel]["C"].append(1.0)
for item in results:
results_tmp = {}
results_tmp = {"n_jobs": [-1], "n_estimators": [100]}
for key, value in results[item].items():
new_key = f"base_estimator__{key}"
try: