Skip to content

Commit 043e89d

Browse files
committed
validation data should be written from anomaly col
1 parent f263813 commit 043e89d

File tree

2 files changed

+3
-5
lines changed

2 files changed

+3
-5
lines changed

ads/opctl/operator/lowcode/anomaly/model/anomaly_dataset.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -36,9 +36,8 @@ def _ingest_data(self, spec):
3636
self.X_valid_dict = dict()
3737
self.y_valid_dict = dict()
3838
for s_id, df in self.get_dict_by_series().items():
39-
print(df, spec.target_column)
40-
self.X_valid_dict[s_id] = df.drop([spec.target_column], axis=1)
41-
self.y_valid_dict[s_id] = df[spec.target_column]
39+
self.X_valid_dict[s_id] = df.drop([OutputColumns.ANOMALY_COL], axis=1)
40+
self.y_valid_dict[s_id] = df[OutputColumns.ANOMALY_COL]
4241

4342

4443
class AnomalyDatasets:

ads/opctl/operator/lowcode/forecast/model/automlx.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -92,8 +92,7 @@ def _build_model(self) -> pd.DataFrame:
9292

9393
# Clean up kwargs for pass through
9494
model_kwargs_cleaned = self.set_kwargs()
95-
time_budget = model_kwargs_cleaned.pop("time_budget", 0)
96-
print(f"found time_budget: {time_budget}")
95+
time_budget = model_kwargs_cleaned.pop("time_budget", -1)
9796

9897
for i, (s_id, df) in enumerate(full_data_dict.items()):
9998
try:

0 commit comments

Comments
 (0)