Можете ли вы сохранить и использовать параметры клиента при сохранении модели catboost с помощью onnx? - PullRequest
1 голос
/ 24 января 2020

У меня есть модель регрессии Catboost, которую я сохраняю с помощью onnx для использования в другом месте, все с использованием Python. Можно ли сохранить пользовательские параметры в модели onnx и извлечь их во время выполнения onnx? Я хочу сохранить метаданные вокруг модели.

1 Ответ

2 голосов
/ 27 января 2020

Пожалуйста, смотрите https://github.com/onnx/onnx/blob/master/onnx/onnx.proto#L249. Вы можете использовать это поле для хранения любых метаданных о модели. Это поле данных в свободной форме; следовательно, onnxruntime (ORT) ничего об этом не знает. Что вы хотите ОРТ извлечь и почему? Также в качестве примечания вы получите более быстрые ответы, если будете использовать https://github.com/Microsoft/onnxruntime/issues. Спасибо.

...