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

Question

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

Answer 1

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