-
Notifications
You must be signed in to change notification settings - Fork 34
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Enable pipeline_ml to share inputs between inference and training pipelines #71
Comments
The concepts of Actually, the post-construction of the pipeline_ml from regular kedro pipeline work well for advanced user, even if it We should consider building a pipeline_ml API in the futur (probably not 0.4.0) that help user building their pipeline_ml using some ml computing concept (fit, predict, transform, ...), kedro-mlflow will translate that in backend to regular kedro pipelines. But for now, i agree, we can just add the possibility to add parameters to inference inputs, kedro-mlflow will pickle them and packge them inside model artifacts at training time. |
I suggest that we don't persist parameters under the hood to avoid side effects: if a user want to use a shared parameter as an input for both training and inference, he must persist it voluntarily (either as the input or output of the shared node). |
We can drop the raise of KedroMlflowPipelineMLDatasetsError , exclusively for "parameters" and "params: xxx" and persist them under the hood. |
Notations:
training
is thekedro.pipeline.Pipeline
object passed as argument "training" of thepipeline_ml
functioninference
is thekedro.pipeline.Pipeline
object passed as argument "inference" of thepipeline_ml
functionCurrently,
inference.inputs()
are forced to be part oftraining.all_outputs()
. however in some situations the two pipelines may share some inputs too. For instance for some NLP models, a processing part include a list of stopwords to remove and these stopwords are inputs (and the same) both fortraining
andinference
.Important points:
PickleDataSet
for instance).The text was updated successfully, but these errors were encountered: