Skip to content

Commit

Permalink
Moving an encoder variable to the train_and_deploy function
Browse files Browse the repository at this point in the history
  • Loading branch information
KirilIakovliev committed Nov 16, 2021
1 parent f997f3d commit 44cd54b
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions hydro_auto_od/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,6 @@


hs_cluster = Cluster(config.cluster_endpoint)
encoder = False

def process_auto_metric_request(training_data_path: str, monitored_model_version_id: int) -> Tuple[int, str]:
logging.info("Started processing auto-od request for modelversion_id=%d", monitored_model_version_id)
Expand Down Expand Up @@ -103,6 +102,7 @@ def train_and_deploy_monitoring_model(monitored_model_version_id: int, training_
monitored_model = ModelVersion.find_by_id(hs_cluster, monitored_model_version_id)
supported_fields: List[ModelField] = TabularOD.get_compatible_fields(monitored_model.signature.inputs)
supported_fields_names: List[str] = sorted([field.name for field in supported_fields])
encoder = False

logging.info(
"Reading training data from %s for modelversion_id=%d",
Expand All @@ -125,7 +125,7 @@ def train_and_deploy_monitoring_model(monitored_model_version_id: int, training_
logging.info(
"Running outlier model selection algorithm for modelversion_id=%d", monitored_model_version_id)
outlier_detector = model_selection(training_data[supported_fields_names])

logging.info(
"Selected an outlier model=%s for modelversion_id=%d",
repr(outlier_detector), monitored_model_version_id)
Expand Down

0 comments on commit 44cd54b

Please sign in to comment.