|
|
@@ -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):
|