Java调用Python训练的xgboost模型预测多分类问题

在角落里123

已于 2024-08-01 09:55:22 修改

阅读量402

点赞数 3

分类专栏： JAVA 文章标签： python java 分类

于 2024-07-31 15:53:45 首次发布

本文链接：https://blog.csdn.net/qq_55996397/article/details/140824448

版权

一、添加依赖

<dependency>
    <groupId>ml.dmlc</groupId>
    <artifactId>xgboost4j_2.12</artifactId>
    <version>2.0.1</version>
</dependency>

二、调用模型进行推理

1.训练出的xgboost模型为一个大型的Json字符串，开头表明了预测所需的参数名称及参数类型，如下如所示

2.调用依赖中提供的方法加载模型并进行预测

private void XGBoostJavaPredictor(Float parameter1, Float parameter2, Float parameter3) throws XGBoostError {
        // 加载XGBoost模型
        Booster model = XGBoost.loadModel(MODEL_PATH);

        // 构造待预测的数据
        float[] features = {parameter1, parameter2, parameter3};
        DMatrix dmatrix = new DMatrix(features, 1,features.length);

        // 进行预测
        float[][] predictions = model.predict(dmatrix);

        // 打印预测结果
        for (float[] prediction : p