机器学习-实战-入门-javaml-KNN处理iris

最新推荐文章于 2024-08-13 06:00:00 发布

橘子派

最新推荐文章于 2024-08-13 06:00:00 发布

阅读量2.6k

点赞数 1

分类专栏：机器学习-实战-Java 文章标签：机器学习 KNN java 人工智能 iris

本文链接：https://blog.csdn.net/sileixinhua/article/details/70660367

版权

机器学习-实战-Java 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

  作者：橘子派 

  声明：版权所有，转载请注明出处，谢谢。 

  源码地址： 
 https://github.com/sileixinhua/javaml_examples_source_code_analysis 

 
 实验环境 

 
 Window10 

 
 Java1.8.0 

  IntelliJ IDEA Community Edition 

  javaml-0.1.4 

用KNN处理iris

package com.silei;

import java.io.File;
import net.sf.javaml.classification.Classifier;
import net.sf.javaml.classification.KNearestNeighbors;
import net.sf.javaml.core.Dataset;
import net.sf.javaml.core.Instance;
import net.sf.javaml.tools.data.FileHandler;

public class Main {
    public static void main(String[] args)throws Exception {
        Dataset data = FileHandler.loadDataset(new File("data/iris.data"), 4, ",");
        Classifier knn = new KNearestNeighbors(5);
        knn.buildClassifier(data);
        Dataset dataForClassification = FileHandler.loadDataset(new File("data/iris.data"), 4, ",");
        int correct = 0, wrong = 0;
        for (Instance inst : dataForClassification) {
            Object predictedClassValue = knn.classify(inst);
            Object realClassValue = inst.classValue();
            if (predictedClassValue.equals(realClassValue))
                correct++;
            else
                wrong++;
        }
        System.out.println("Correct predictions  " + correct);
        System.out.println("Wrong predictions " + wrong);
    }
}