Exemplo de inferência de modelo
Este Notebook usa um modelo ElasticNet treinado no diabetes dataset descrito no Track scikit-learn model treinamento with MLflow. Este Notebook mostra como fazer:
- Selecione um modelo a ser implantado usando a UI do experimento MLflow
- Carregar o modelo treinado como um modelo do scikit-learn
- Criar um PySpark UDF a partir do modelo
- Aplicar o UDF para adicionar uma coluna de previsão a um DataFrame