Bladeren bron

Added support for wildcard feature.

Tom Flucke 6 jaren geleden
bovenliggende
commit
8ca14108ad
1 gewijzigde bestanden met toevoegingen van 6 en 1 verwijderingen
  1. 6 1
      src/classifiers/Vector.py

+ 6 - 1
src/classifiers/Vector.py

@@ -30,7 +30,12 @@ class FeatureVector:
     # ~~ex: For 2nd, 4th and 5th features to be active, pass [0,1,0,1,1]~~
     # ex: ["total_time", "average_iat", "dead_time"]
     def set_features(self, features: typing.List[str]):
-        self.activefeatures = [self.sampleInfo[feature] for feature in features]
+        if "*" in features:
+            self.activefeatures = [self.sampleInfo[feature]
+                                   for feature in self.sampleInfo.keys()]
+        else:
+            self.activefeatures = [self.sampleInfo[feature]
+                                   for feature in features]
         return self
     
     def get(self):