纯Java实现的XGBoost预测器:xgboost-predictor-java

1. 项目基础介绍及编程语言

xgboost-predictor-java 是一个开源项目,提供了一个纯Java实现的XGBoost预测器,适用于在线预测任务。该项目主要使用Java编程语言开发,同时也包含Scala和Groovy等语言的部分代码。

2. 核心功能

项目的核心功能是提供了一个能够加载XGBoost模型并对其进行预测的Java库。主要特点包括:

  • 模型加载:支持加载XGBoost模型文件。
  • 特征向量处理:支持从数组或映射中创建密集或稀疏的特征向量。
  • 预测功能:能够进行概率预测、分类预测、输出预测边际以及预测每个树中的叶索引。

3. 最近更新的功能

最近更新的功能主要包括:

  • 性能优化:对模型加载和预测的效率进行了优化,提高了预测任务的吞吐量。
  • 支持更多的目标函数和模型类型:增加了对更多目标函数的支持,如gblineargbtreedart等。
  • API增强:增强了API的功能,包括预测概率、输出边际以及预测叶索引等。

以上更新使得xgboost-predictor-java在性能和功能上都有了显著的提升,为Java开发者提供了一个更加高效、易用的XGBoost预测工具。

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐