如何使用数据预处理（文本数据）部署 mlflow 模型

Question

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

Answer 1

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