将gensim LDA模型保存到ONNX

时间:2020-03-03 20:33:16

标签: gensim lda onnx

是否可以将gensim LDA模型保存为ONNX格式?我们需要能够使用Python / gensim进行训练,然后将其实现为Onnx模型以进行发布和使用。

1 个答案:

答案 0 :(得分:0)

当前(2020年3月,gensim-3.8.1)我不知道gensim中对ONNX格式的任何内置支持。

提供的ONNX格式可以很好地表示LDA模型-&here's an indiction it does-这将是一个合理的新功能。

您可以在gensim issue tracker处添加功能请求,但是要添加该功能,可能需要需要该功能的熟练开发人员的帮助,并且可以编写代码和测试用例。 / p>