Можно ли получить дерево solution_path из вызовов моделей sklearn, сохраненных с помощью skl2onnx? - PullRequest
0 голосов
/ 17 января 2020

Например, следующее: https://github.com/onnx/sklearn-onnx/blob/master/docs/tutorial.rst

Но с

model = sklearn.ensemble.RandomForestRegressor()

Мы можем получить

models.estimators_[0].decision_path(X)

Можем ли мы сделать это с моделью, преобразованной в ONNX?

Или есть способ просто загрузить файл ONNX обратно в sklearn?

1 Ответ

0 голосов
/ 05 февраля 2020

Это, к сожалению, невозможно - конвертер skl2onnx не предоставляет пути принятия решения в преобразованной модели ONNX. В случае с регрессорами SciKitLearn мы представляем только прогнозируемый результат в модели ONNX. В классификаторах мы представляем прогнозируемый класс и оценки вероятности / функции решения.

Также нет способа взять преобразованную модель ONNX и загрузить ее обратно в SciKitLearn.

...