如何将XGBoost/LightGBM模型保存到PostgreSQL数据库中，以便在Java中进行后续推理？
如何将XGBoost/LightGBM模型保存到PostgreSQL数据库中，以便在Java中进行后续推理？
提问于 2022-12-02 13:28:06
我被限制为模型本身或相关组件(系数，..)的“模型存储”( PostgreSQL )。显然，PostgreSQL还远远不是一个成熟的模型存储，因此我不能排除必须在Java中实现整个模型培训过程的可能性。
我找不到一个将PostgreSQL数据库作为模型中间存储的解决方案。直接将文件写入磁盘/其他存储库对我来说并不是一种选择。我考虑过从Python应用程序中调用Java代码，但我不知道这是否是后续推理任务和以后的有效解决方案.是否有方法序列化PMML或其他可以通过算法的Java实现加载的格式？或直接使用模型定义/参数复制模型的方法.
回答 1
Stack Overflow用户
发布于 2022-12-02 18:10:34
使用PostgreSQL作为虚拟模型存储：
Python.
- Establish
- 在PostgreSQL连接中训练模型，将模型以Pickle数据格式转储到"models“表。显然，主列的数据类型应该是BLOB.
- ，只要您想要将该模型用于某些应用程序，请将其从“模型”表.
中提取出来。
"models“表可能有额外的列，用于以其他数据格式(如PMML )存储模型。假设您使用了正确的Python到PMML转换工具，您可以假设相同模型的Pickle表示和PMML表示在功能上是相同的(即。当给定相同的输入时，做出相同的预测)。在Java/JVM应用程序中使用PMML很容易。
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/74656521
复制相关文章
相似问题
腾讯云开发者
