Exemplo de inferência de modelo
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