Browse Source

remove of warning from sklearn

Jérôme BUISINE 1 year ago
parent
commit
a201cba7b5
1 changed files with 1 additions and 4 deletions
  1. 1 4
      find_best_attributes_surrogate.py

+ 1 - 4
find_best_attributes_surrogate.py

@@ -48,10 +48,7 @@ from optimization.callbacks.SurrogateMonoCheckpoint import SurrogateMonoCheckpoi
 models_list         = cfg.models_names_list
 
 from warnings import simplefilter
-from sklearn.exceptions import ConvergenceWarning, DeprecationWarning, VisibleDeprecationWarning
-simplefilter("ignore", category=ConvergenceWarning)
-simplefilter("ignore", category=DeprecationWarning)
-simplefilter("ignore", category=VisibleDeprecationWarning)
+simplefilter("ignore")
 
 # default validator
 def validator(solution):