Merge pull request #394 from ZJUEarthData/fix/ano

Fix/ano
ZJUEarthData · Oct 9, 2024 · eb0a18b · eb0a18b
2 parents 0ed77d1 + f17269d
commit eb0a18b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/geochemistrypi/data_mining/model/detection.py b/geochemistrypi/data_mining/model/detection.py
@@ -20,7 +20,7 @@
 class AnomalyDetectionWorkflowBase(WorkflowBase):
     """The base workflow class of anomaly detection algorithms."""
 
-    # common_function = []
+    common_function = [func.value for func in AnormalyDetectionCommonFunction]
 
     def __init__(self) -> None:
         super().__init__()
@@ -153,7 +153,7 @@ class IsolationForestAnomalyDetection(AnomalyDetectionWorkflowBase):
     """The automation workflow of using Isolation Forest algorithm to make insightful products."""
 
     name = "Isolation Forest"
-    # special_function = []
+    special_function = []
 
     def __init__(
         self,
@@ -306,7 +306,7 @@ class LocalOutlierFactorAnomalyDetection(AnomalyDetectionWorkflowBase):
     """The automation workflow of using Local Outlier Factor algorithm to make insightful products."""
 
     name = "Local Outlier Factor"
-    # special_function = []
+    special_function = [func.value for func in LocalOutlierFactorSpecialFunction]
 
     def __init__(
         self,