From 32a82ceb3c4a23b3a2c762d7be00725c64e7b910 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ricardo=20Montan=CC=83ana?= Date: Thu, 10 Mar 2022 12:07:09 +0100 Subject: [PATCH] Add njobs and nestimators to build_grid --- results/grid_input_accuracy_ODTE.json | 24 ++++++++++++++++++++++++ src/build_grid.py | 2 +- 2 files changed, 25 insertions(+), 1 deletion(-) 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: