使用 onnx 保存 catboost 模型时，是否可以保存和使用客户参数?

Question

我有一个Catboost回归模型，我用onnx保存了它，以便在其他地方使用，所有这些都使用Python。是否可以将自定义参数保存在 onnx 模型中并使用 onnx 运行时提取它们？我想保存模型周围的元数据。

Answer 1

请参阅 https://github.com/onnx/onnx/blob/master/onnx/onnx.proto#L249。您可以使用此字段存储有关模型的任何元数据。这是一个自由格式的数据字段;因此onnxruntime (ORT(对此一无所知。您希望ORT提取什么，为什么？另外作为旁注，如果您使用 https://github.com/Microsoft/onnxruntime/issues，您将获得更快的响应。谢谢。