modify: jupyter输出优化

feature/filter_strategy_base.py

@@ -5,9 +5,10 @@
 @desc: 特征筛选基类
 """
 import abc
-from typing import Dict, List
+from typing import Dict
 
 from entitys import DataProcessConfigEntity, DataPreparedEntity, CandidateFeatureEntity, MetricFucEntity
+from init import warning_ignore
 
 
 class FilterStrategyBase(metaclass=abc.ABCMeta):
@@ -20,10 +21,7 @@ class FilterStrategyBase(metaclass=abc.ABCMeta):
 
         jupyter = self._data_process_config.jupyter
         if jupyter:
-            from pandas.core.common import SettingWithCopyWarning
-            import warnings
-            warnings.simplefilter(action="ignore", category=SettingWithCopyWarning)
-            warnings.simplefilter(action="ignore", category=RuntimeWarning)
+            warning_ignore()
 
     @property
     def data_process_config(self):

init/__init__.py

@@ -11,7 +11,7 @@ matplotlib.use('Agg')
 
 import matplotlib.pyplot as plt
 
-__all__ = ['init']
+__all__ = ['init', 'warning_ignore']
 
 
 def init():
@@ -21,5 +21,12 @@ def init():
     plt.rcParams['figure.max_open_warning'] = 1000
 
 
+def warning_ignore():
+    from pandas.core.common import SettingWithCopyWarning
+    import warnings
+    warnings.simplefilter(action="ignore", category=SettingWithCopyWarning)
+    warnings.simplefilter(action="ignore", category=RuntimeWarning)
+
+
 if __name__ == "__main__":
     pass