Minors

Author: jhosoume

analysis/compare_filters.py

@@ -12,7 +12,7 @@
 from NoiseFiltersPy._injectors import _implemented_injectors 
 
 DATASETS_PATH = "analysis/datasets/"
-FILTERS = ["TomekLinks", "CNN", "DROP", "ENN", "AENN"]
+FILTERS = ["HARF", "TomekLinks", "CNN", "DROP", "ENN", "AENN"]
 
 datasets = [f for f in listdir(DATASETS_PATH)
                 if ( isfile(join(DATASETS_PATH, f)) and
@@ -41,8 +41,6 @@ def calculate_filter_f1(dataset, filter, injector, rate = 0.1):
             )
         attrs = ct.fit_transform(attrs)
 
-    attrs = attrs.values
-
     injector = injector(attrs, target, rate)
     injector.generate()
 
@@ -64,4 +62,4 @@ def calculate_filter_f1(dataset, filter, injector, rate = 0.1):
         results[injector] = dict(map(calculate_f1, datasets))
 
     results = pd.DataFrame(results)
-    results.to_csv("analysis/compare_filters/{}.csv".format(filter))
+    results.to_csv("analysis/compare_filters/{}.csv".format(filter))

requirements.txt

@@ -1,4 +1,5 @@
 numpy
 scipy
 scikit-learn
+pandas
 gower