【问题标题】:Is it possible to get tree decision_path from calls to sklearn models saved with skl2onnx?是否可以从调用 skl2onnx 保存的 sklearn 模型中获取树决策路径?
【发布时间】:2020-01-17 11:52:20
【问题描述】:

例如:https://github.com/onnx/sklearn-onnx/blob/master/docs/tutorial.rst

但是有

model = sklearn.ensemble.RandomForestRegressor()

我们可以搞定

models.estimators_[0].decision_path(X)

我们可以使用转换为 ONNX 的模型来做到这一点吗?

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

【问题讨论】:

    标签: python scikit-learn onnx


    【解决方案1】:

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

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

    【讨论】:

      猜你喜欢
      • 2018-12-09
      • 2016-11-15
      • 2021-10-01
      • 2014-11-26
      • 2016-10-13
      • 2015-12-29
      • 2020-01-09
      • 2023-03-12
      • 2018-08-14
      相关资源
      最近更新 更多