Преглед на файлове

remove of warning from sklearn

Jérôme BUISINE преди 3 години
родител
ревизия
a201cba7b5
променени са 1 файла, в които са добавени 1 реда и са изтрити 4 реда
  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):