draft commit rcf

oracle · Aug 22, 2024 · 8dbccae · 8dbccae
1 parent 97c78b0
commit 8dbccae
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/ads/opctl/operator/lowcode/anomaly/model/randomcutforest.py b/ads/opctl/operator/lowcode/anomaly/model/randomcutforest.py
@@ -33,7 +33,6 @@ def _build_model(self) -> AnomalyOutput:
         # self.outlier_map = {1: 0, -1: 1}
 
         anomaly_output = AnomalyOutput(date_column="index")
-        # TODO: PDB
 
         # Set tree parameters
         num_trees = model_kwargs.get("num_trees", 200)
@@ -42,8 +41,11 @@ def _build_model(self) -> AnomalyOutput:
 
         for target, df in self.datasets.full_data_dict.items():
             df_values = df[self.spec.target_column].astype(float).values
+
+            # TODO: Update size to log logic
             points = np.vstack(list(rrcf.shingle(df_values, size=4)))
 
+            # TODO: remove hardcode
             sample_size_range = (1, 6)
             n = points.shape[0]
             avg_codisp = pd.Series(0.0, index=np.arange(n))
@@ -62,16 +64,19 @@ def _build_model(self) -> AnomalyOutput:
                 np.add.at(index, codisp.index.values, 1)
 
             avg_codisp /= index
+            # TODO: remove hardcode
             avg_codisp.index = df.iloc[(4 - 1) :].index
             avg_codisp = (avg_codisp - avg_codisp.min()) / (
                 avg_codisp.max() - avg_codisp.min()
             )
 
+            # TODO: use model kwargs for percentile threshold
             y_pred = (avg_codisp > np.percentile(avg_codisp, 95)).astype(int)
 
-            import pdb
+            # TODO: rem pdb
+            # import pdb
 
-            pdb.set_trace()
+            # pdb.set_trace()
             print("Done")
 
             # scores = model.score_samples(df)