From 24ce5ba9eee53f0e0f1e66e13a7f90351b4d0cf4 Mon Sep 17 00:00:00 2001
From: Joaquin Torres Bravo <joaquintobrw@gmail.com>
Date: Mon, 3 Jun 2024 12:34:55 +0200
Subject: [PATCH] Regen hyperpameters with XGB relaxation

---
 model_selection/hyperparam_tuning.py | 2 --
 1 file changed, 2 deletions(-)

diff --git a/model_selection/hyperparam_tuning.py b/model_selection/hyperparam_tuning.py
index 3fb508d..0892526 100644
--- a/model_selection/hyperparam_tuning.py
+++ b/model_selection/hyperparam_tuning.py
@@ -147,8 +147,6 @@ if __name__ == "__main__":
             hyperparam_df = pd.DataFrame(index=list(models.keys()), columns=['Best Parameters'])
             for model_name, model in models.items():
                 print(f"{group}-{method_names[j]}-{model_name}")
-                if model_name != 'XGB':
-                    continue
                 # Find optimal hyperparams for curr model
                 params = hyperparameters[model_name]
                 search = RandomizedSearchCV(model, param_distributions=params, cv=cv, n_jobs=10, scoring='precision')
-- 
2.24.1