REFACTOR: Deleted non-working criterion

denBruneBarone · Mar 15, 2024 · b5821df · b5821df
1 parent f07f10d
commit b5821df
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/machine_learning/config.py b/machine_learning/config.py
@@ -6,7 +6,7 @@ class HPConfig:
 
 class GridSearchConfig:
  param_grid = {
- 'criterion': ['friedman_mse', 'absolute_error', 'squared_error'],
+ 'criterion': ['friedman_mse', 'squared_error'],
  'max_depth': [2, 3, 4, 5, 6, 7, 8],
  'max_features': [None, 'sqrt', 'log2'],
  'max_leaf_nodes': [2, 3, 4, 5, 6, 7, 8, 9, 10]