Use weka LinearRegression Regression - Java Machine Learning AI

Java examples for Machine Learning AI:weka

  1. HOME
  2. Java
  3. Machine Learning AI
  4. weka

Description

Use weka LinearRegression Regression

Demo Code

import weka.classifiers.Evaluation;
import weka.classifiers.functions.LinearRegression;
import weka.classifiers.functions.SMOreg;
import weka.core.Instances;
import weka.core.converters.ConverterUtils.DataSource;

public class Regression {
    public static void main(String[] args) throws Exception {
        DataSource source = new DataSource("house.arff");
        Instances dataset = source.getDataSet();
        dataset.setClassIndex(dataset.numAttributes() - 1);
        /**// w w w  .jav a  2s  . c o m
         * linear regression model
         */
        LinearRegression lr = new LinearRegression();
        lr.buildClassifier(dataset);
        //System.out.println(lr);
        Evaluation lreval = new Evaluation(dataset);
        lreval.evaluateModel(lr, dataset);
        System.out.println(lreval.toSummaryString());
        /**
         * svm regression model
         */
        SMOreg smoreg = new SMOreg();
        smoreg.buildClassifier(dataset);
        Evaluation svmregeval = new Evaluation(dataset);
        svmregeval.evaluateModel(smoreg, dataset);
        System.out.println(svmregeval.toSummaryString());

    }
}

Related Tutorials