How to create seperate mlflow custom models for training and prediction?

Question

My requirement is to create separate Mlflow custom models for training and prediction. I want to create training model and use those training model in prediction model

Answer 1

You can use mlflow.pyfunc to create a custom model.

You can find additional details of mlflow.pyfunc on this link.

This link will also help you get a overall view on models