diff --git a/results/grid_input_accuracy_ODTE.json b/results/grid_input_accuracy_ODTE.json index 879927a..332abc1 100644 --- a/results/grid_input_accuracy_ODTE.json +++ b/results/grid_input_accuracy_ODTE.json @@ -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, diff --git a/src/build_grid.py b/src/build_grid.py index a9a2f96..b53c00c 100755 --- a/src/build_grid.py +++ b/src/build_grid.py @@ -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: