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

Notebook de inferência MLflow

Abra o bloco de anotações em outra guia