Yggdrasil决策森林:一个快速且可扩展的决策森林库

摘要:Yggdrasil决策森林是一个用于训练、服务和解释决策森林模型的库,针对研究和生产工作,用C++实现,并提供C++、命令行界面、Python(名为TensorFlow决策森林)、JavaScript、Go和Google Sheets(名为Sheets的简单ML)版本。该库自2018年以来逐步发展,遵循机器学习库和框架的四个设计原则:易用性、安全性、模块化和高级抽象以及与其他机器学习库的集成。本文详细描述了这些原则,并展示了它们如何指导库的设计。然后我们展示了我们的库在一组经典机器学习问题上的应用。最后,我们报道了与相关解决方案的基准测试结果。

作者:Mathieu Guillame-Bert, Sebastian Bruch, Richard Stotz, Jan Pfeifer

论文ID:2212.02934

分类:Machine Learning

分类简称:cs.LG

提交时间:2023-08-10

PDF 下载: 英文版 中文版pdf翻译中