Minors

Author: jhosoume

NoiseFiltersPy/Injector.py

@@ -26,7 +26,7 @@ def __init__(self, attributes, labels, rate: float = 0.1) -> None:
         else:
             self._attrs = attributes
 
-        if not isinstance(attributes, pd.DataFrame):
+        if not isinstance(labels, pd.DataFrame):
             self._labels = pd.DataFrame(labels)
         else:
             self._labels = labels

analysis/compare_filters.py

@@ -40,6 +40,7 @@ def calculate_filter_f1(dataset, filter, injector, rate = 0.1):
             remainder = "passthrough"
             )
         attrs = ct.fit_transform(attrs)
+    attrs = np.array(attrs)
 
     injector = injector(attrs, target, rate)
     injector.generate()