Weka生成和加载PMML文件

本文介绍了如何使用Weka训练数据分类模型并以PMML格式导出和加载。通过示例代码展示了从训练模型到导出PMML文件,再到加载模型和验证模型的过程。此外,还提到了另一种模型保存方法——Java序列化。
摘要由CSDN通过智能技术生成

网络上太多示例展示了Weka怎么样调用数据分类算法,但想想我如何针对一个训练好的分类模型进行重用呢。所以必须要“导出来”。导出模型,一个标准的方式就是用PMML了。

标准的方式,Weka一定会支持。找了一下,不负所望。将上一次的代码整理了一下,加上PMML的支持:

训练模型,导出PMML文件

    static void trainModelToPmml(String arffFile, int classIndex, File pmmlFile) throws Exception {

        File inputFile = new File(arffFile); //训练文件
        ArffLoader loader = new ArffLoader();
        loader.setFile(inputFile);
        Instances insTrain = loader.getDataSet(); // 读入训练文件
        insTrain.setClassIndex(classIndex);

        Logistic logic=new Logistic();
        logic.buildClassifier(insTrain);//根据训练数据构造分类器

        String pmml = logic.toPMML(insTrain);

        PrintWriter writer = new PrintWriter(new BufferedOutputStream(new FileOutputStream(pmmlFile)));
        writer.println(pmml);
        writer.close();
    }

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值