具有定制损失函数的随机森林

时间:2014-04-15 08:29:02

标签: machine-learning scikit-learn random-forest

我是机器学习领域的初学者。对于项目,我必须在随机森林分类中使用自定义损失函数。我到现在一直使用scikit。通过scikit实现这一点的建议会更有帮助。

1 个答案:

答案 0 :(得分:3)

损失函数(基于分类树的基尼杂质和熵)在scikit的_tree.pyx cython文件中实现(它们在源中称为标准)。您可以从修改/添加这些功能开始。如果将自定义丢失函数(标准)添加到cython文件中,还需要在tree.py python文件中公开它(查看CRITERIA_CLF和CRITERIA_REG列表)。