diff --git a/nodes/flex/features_proximity.py b/nodes/flex/features_proximity.py index 65cbd3d..2b941d4 100644 --- a/nodes/flex/features_proximity.py +++ b/nodes/flex/features_proximity.py @@ -14,6 +14,9 @@ class ProximityFeature(BaseFeature): self.proximity_values = None self.normalization_method = normalization_method + def get_extraction_methods(self): + return ["normalization_method"] + def extract(self): from scipy.spatial.distance import cdist