是否可以从调用使用 skl2onnx 保存的 sklearn 模型中获取树decision_path?

Question

例如以下内容：https://github.com/onnx/sklearn-onnx/blob/master/docs/tutorial.rst

但是随着

model = sklearn.ensemble.RandomForestRegressor()

我们可以掌握

models.estimators_[0].decision_path(X)

我们可以使用转换为 ONNX 的模型来执行此操作吗？

或者有没有办法简单地将ONNX文件加载回sklearn？

Answer 1

不幸的是，这是不可能的 - skl2onnx 转换器不会在转换后的 ONNX 模型中公开决策路径。对于SciKitLearn的回归器，我们只在ONNX模型中公开预测的分数。在分类器中，我们公开了预测类和概率/决策函数分数。

也没有办法将转换后的ONNX模型加载回SciKitLearn。