我最近开始学习 MLflow，我正在一个简单的猫狗分类器上尝试 ML 生命周期。这是代码。
我已经正确设置了使用服务的模型mlflow serve。但是，有一个问题。我已将模型签名配置为采用特定形状的输入，并且 API 端点在发送请求时将采用一个数组。
在send_request.py文件中，我正在对 CNN 输入执行预处理，然后将 Numpy 数组作为列表发送。我想做的是，为模型设置我自己的自定义预测函数。这样端点只会将原始图像作为输入而不是数组。
我按照本教程进行操作。但不能使它适用于我的情况。
我了解在记录/保存模型时，我需要提供自定义对象。但我不确定如何获取在这些自定义模型中执行预测的模型对象。对于这种特殊情况，没有多少资源可供参考。