use linear Regression coefficients - Java Machine Learning AI

Java examples for Machine Learning AI:weka

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

Description

use linear Regression coefficients

Demo Code



import java.io.BufferedReader;
import java.io.FileNotFoundException;
import java.io.FileReader;
import java.io.IOException;
import java.io.Reader;

import weka.classifiers.functions.LinearRegression;
import weka.core.Instances;

public class Weka {
    public static void main(String[] args) {
        String inputFile = "bank-test.arff";
        Reader r;//ww w  .j  a v a 2  s . c o  m

        try {
            r = new BufferedReader(new FileReader(inputFile));

            Instances instances = new Instances(r);
            instances.setClassIndex(instances.numAttributes() - 1);

            LinearRegression linearRegression = new LinearRegression();
            linearRegression.buildClassifier(instances);

            double[] coef = linearRegression.coefficients();
            double myHouseValue = (coef[0] * 3198) + (coef[1] * 9669)
                    + (coef[2] * 5) + (coef[3] * 3) + (coef[4] * 1)
                    + coef[6];

            System.out.println(myHouseValue);
        } catch (FileNotFoundException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}

Related Tutorials