Exemplo de inferência de modelo

Este Notebook usa um modelo ElasticNet treinado no dataset de diabetes descrito em Track Scikit-Learn model training with MLflow. Este Notebook mostra como:

• Selecione um modelo para aprimorar usando a interface do usuário do experimento MLflow

• Carregue o modelo treinado como um modelo Scikit-Learn

• Crie um UDF PySpark a partir do modelo

• Aplique o UDF para adicionar uma coluna de previsão a um DataFrame

Notebook de inferência MLflow

Abra o bloco de anotações em outra guia Copiar link para importação