Просмотр исходного кода

remove of warning from sklearn

Jérôme BUISINE 3 лет назад
Родитель
Сommit
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):