ComeçarComece gratuitamente

Criando uma classe Python personalizada

MLO fluxo oferece uma maneira de criar modelos personalizados para que você possa oferecer suporte a uma ampla variedade de casos de uso. Para criar modelos personalizados, o fluxo MLpermite que os usuários criem uma classe Python que herda a classe mlflow.pyfunc.PythonModel. A classe PythonModel oferece personalização, fornecendo métodos para lógica de inferência personalizada e dependências de artefatos.

Neste exercício, você criará uma nova classe Python para um modelo personalizado que carrega um modelo específico e, em seguida, decodifica os rótulos após a inferência. O módulo mlflow será importado.

Este exercício faz parte do curso

Introdução ao fluxo ML

Ver Curso

Instruções de exercício

  • Crie uma classe Python com o nome CustomPredict.
  • Defina o método load_context() usado para carregar artefatos em uma classe personalizada.
  • Defina o método predict() para definir a inferência personalizada.

Exercício interativo prático

Experimente este exercício preenchendo este código de exemplo.

# Create Python Class
class ____(mlflow.pyfunc.PythonModel):
    # Set method for loading model
    def ____(self, context):
        self.model = mlflow.sklearn.load_model("./lr_model/")
    # Set method for custom inference     
    def ____(self, context, model_input):
        predictions = self.model.predict(model_input)
        decoded_predictions = []  
        for prediction in predictions:
            if prediction == 0:
                decoded_predictions.append("female")
            else:
                decoded_predictions.append("male")
        return decoded_predictions
Editar e executar código