Skip to content

Commit

Permalink
Update custom method to not include the module name
Browse files Browse the repository at this point in the history
Signed-off-by: Sunish Sheth <[email protected]>
  • Loading branch information
sunishsheth2009 authored and Jin Zhang committed Nov 10, 2022
1 parent 5fb26ff commit ddc5dc6
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions regression/recipe.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -23,14 +23,14 @@ steps:
split_ratios: {{SPLIT_RATIOS|default([0.75, 0.125, 0.125])}}
# Specifies the method to use to perform additional cleaning on split datasets
# Note that arbitrary transformations should go into the transform step
post_split_filter_method: steps.split.create_dataset_filter
post_split_filter_method: create_dataset_filter
transform:
# Specifies the method that defines the data transformations to apply during model inference
transformer_method: steps.transform.transformer_fn
transformer_method: transformer_fn
train:
using: custom
# Specifies the method that defines the estimator type and parameters to use for model training
estimator_method: steps.train.estimator_fn
estimator_method: estimator_fn
evaluate:
# Sets performance thresholds that a trained model must meet in order to be eligible for
# registration to the MLflow Model Registry
Expand Down

0 comments on commit ddc5dc6

Please sign in to comment.