python - 是否可以从调用 skl2onnx 保存的 sklearn 模型中获取树决策路径？

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？

标签： pythonscikit-learnonnx

Answer 1

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

也无法获取转换后的 ONNX 模型并将其加载回 SciKitLearn。