critical FS bug fixes

firefly-cpp · Dec 5, 2020 · b64445a · b64445a
1 parent 7d9c4fa
commit b64445a
Show file tree

Hide file tree

Showing 6 changed files with 7 additions and 2 deletions.
diff --git a/niaaml/preprocessing/feature_selection/bat_algorithm.py b/niaaml/preprocessing/feature_selection/bat_algorithm.py
@@ -6,6 +6,7 @@
 from niaaml.preprocessing.feature_selection.feature_selection_algorithm import FeatureSelectionAlgorithm
 from niaaml.utilities import ParameterDefinition, MinMax
 from niaaml.preprocessing.feature_selection.utility import _FeatureSelectionThresholdBenchmark
+import numpy
 
 __all__ = [
     'BatAlgorithm'

diff --git a/niaaml/preprocessing/feature_selection/differential_evolution.py b/niaaml/preprocessing/feature_selection/differential_evolution.py
@@ -6,6 +6,7 @@
 from niaaml.preprocessing.feature_selection.feature_selection_algorithm import FeatureSelectionAlgorithm
 from niaaml.utilities import ParameterDefinition, MinMax
 from niaaml.preprocessing.feature_selection.utility import _FeatureSelectionThresholdBenchmark
+import numpy
 
 __all__ = [
     'DifferentialEvolution'

diff --git a/niaaml/preprocessing/feature_selection/grey_wolf_optimizer.py b/niaaml/preprocessing/feature_selection/grey_wolf_optimizer.py
@@ -6,6 +6,7 @@
 from niaaml.preprocessing.feature_selection.feature_selection_algorithm import FeatureSelectionAlgorithm
 from niaaml.utilities import ParameterDefinition, MinMax
 from niaaml.preprocessing.feature_selection.utility import _FeatureSelectionThresholdBenchmark
+import numpy
 
 __all__ = [
     'GreyWolfOptimizer'

diff --git a/niaaml/preprocessing/feature_selection/jDEFSTH.py b/niaaml/preprocessing/feature_selection/jDEFSTH.py
@@ -5,6 +5,7 @@
 from sklearn.linear_model import LogisticRegression
 from niaaml.preprocessing.feature_selection.feature_selection_algorithm import FeatureSelectionAlgorithm
 from niaaml.preprocessing.feature_selection.utility import _FeatureSelectionThresholdBenchmark
+import numpy
 
 __all__ = [
     'jDEFSTH'

diff --git a/niaaml/preprocessing/feature_selection/particle_swarm_optimization.py b/niaaml/preprocessing/feature_selection/particle_swarm_optimization.py
@@ -1,11 +1,10 @@
 import sys
 from NiaPy.algorithms.basic import ParticleSwarmOptimization as PSO
 from NiaPy.task import StoppingTask
-from sklearn.model_selection import train_test_split
 from sklearn.linear_model import LogisticRegression
 from niaaml.preprocessing.feature_selection.feature_selection_algorithm import FeatureSelectionAlgorithm
 from niaaml.utilities import ParameterDefinition, MinMax
-import numpy as np
+import numpy
 from niaaml.preprocessing.feature_selection.utility import _FeatureSelectionThresholdBenchmark
 
 __all__ = [

diff --git a/niaaml/preprocessing/feature_selection/utility.py b/niaaml/preprocessing/feature_selection/utility.py
@@ -1,6 +1,8 @@
 from niaaml.utilities import Factory
 import niaaml.preprocessing.feature_selection as fs
 from NiaPy.benchmarks import Benchmark
+from sklearn.model_selection import train_test_split
+from sklearn.linear_model import LogisticRegression
 
 __all__ = [
     'FeatureSelectionAlgorithmFactory',