catboost - 使用 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 ，您将获得更快的响应。谢谢。