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
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