如何部署带有数据预处理的mlflow模型(文本数据)

Question

我开发了keras文本分类模型。我有预处理的数据(标记化)。我已成功记录训练好的模型(mlflow.keras.log_model)。我已经使用mlflow服务为模型服务了。现在，在对文本数据进行预测时，我需要使用用于训练的相同标记器对象进行预处理。如何预处理测试数据并从服务模型中获得预测。

Answer 1

您可以记录一个自定义的python模型：https://www.mlflow.org/docs/latest/models.html#custom-python-models