【发布时间】:2022-05-19 20:51:53
【问题描述】:
我开发了 keras 文本分类模型。我已经预处理数据(标记化)。我已经成功记录了训练好的模型(mlflow.keras.log_model)。我已经使用 mlflow serve 为模型提供服务。现在,在对文本数据进行预测时,我需要使用用于训练的相同标记器对象进行预处理。 如何预处理测试数据并从服务模型中获取预测。
【问题讨论】:
标签: mlflow
我开发了 keras 文本分类模型。我已经预处理数据(标记化)。我已经成功记录了训练好的模型(mlflow.keras.log_model)。我已经使用 mlflow serve 为模型提供服务。现在,在对文本数据进行预测时,我需要使用用于训练的相同标记器对象进行预处理。 如何预处理测试数据并从服务模型中获取预测。
【问题讨论】:
标签: mlflow
您可以记录自定义 python 模型: https://www.mlflow.org/docs/latest/models.html#custom-python-models
【讨论】: