是否可以在scikit-leanr决策树中设置拆分值的精度？

Question

当在决策树中访问节点的tree_.threshold时，类型似乎是float。有没有办法将阈值的“精度”设置为整数？ 在我的情况下，这些特征是所有的整数值，所以不需要在以这种精度定义的值上分割。

Answer 1

不，scikit-learn树在编译时使用double类型的阈值（请参阅https://github.com/scikit-learn/scikit-learn/blob/master/sklearn/tree/_tree.pyx）。固定的数据类型允许生成高效的C代码。您必须更改scikit-learn源代码，从Cython .pyx文件重建C扩展，然后安装修改后的scikit-learn以使用整数阈值。