examples.OrdinaryLeastSquares Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of RCaller Show documentation
http://www.mhsatman.com
The newest version!
package examples;

import com.github.rcaller.rstuff.RCaller;
import com.github.rcaller.rstuff.RCode;

import java.util.logging.Level;
import java.util.logging.Logger;

/**
 *
 * @author Mehmet Hakan Satman
 * @since 2.0
 * @version 2.0
 * 
 */
public class OrdinaryLeastSquares {
  
  public static void main(String[] args) {
    new OrdinaryLeastSquares();
  }

  /**
   *
   * Ordinary Least Squares with RCaller.
   * This class runs the lm() function of R
   * for regressing user defined vector y on vector x from java
   */
  public OrdinaryLeastSquares() {
    
    try {
      /**
       * Creating an instance of RCaller class
       */
      RCaller caller = RCaller.create();
      RCode code = RCode.create();

      /**
       * Creating vectors x and y
       */
      double[] x = new double[]{1, 2, 3, 4, 5, 6, 7, 8, 9, 10};
      double[] y = new double[]{2, 4, 6, 8, 10, 12, 14, 16, 18, 30};

      /**
       * Converting Java arrays to R arrays
       */
      code.addDoubleArray("x", x);
      code.addDoubleArray("y", y);

      /**
       * R code for regression of y on x
       */
      code.addRCode("ols<-lm(y~x)");

      /**
       * Run all! Our regression code returns someting
       * and we want to handle the variable 'ols'
       */
      caller.setRCode(code);
      caller.runAndReturnResult("ols");

      /**
       * Getting names of components of the variable 'ols'
       */
      System.out.println("Available results from lm() object:");
      System.out.println(caller.getParser().getNames());


      /**
       * Getting ols$residulas, ols$coefficients and ols$fitted.values
       * The names after '$' are components of the lm() object in R language
       */
      double[] residuals = caller.getParser().getAsDoubleArray("residuals");
      double[] coefficients = caller.getParser().getAsDoubleArray("coefficients");
      double[] fitteds = caller.getParser().getAsDoubleArray("fitted_values");

      /**
       * Printing results
       */
      System.out.println("Coefficients:");
      for (int i = 0; i < coefficients.length; i++) {
        System.out.println("Beta " + i + " = " + coefficients[i]);
      }
      
      System.out.println("Residuals:");
      for (int i = 0; i < residuals.length; i++) {
        System.out.println(i + " = " + residuals[i]);
      }

      System.out.println("Fitted Values:");
      for (int i = 0; i < fitteds.length; i++) {
        System.out.println(i + " = " + fitteds[i]);
      }
      
    } catch (Exception e) {
      /**
       * Note that, RCaller does some OS based works such as creating an external process and
       * reading files from temporary directories or creating images for plots. Those operations
       * may cause exceptions for those that user must handle the potential errors. 
       */
      Logger.getLogger(OrdinaryLeastSquares.class.getName()).log(Level.SEVERE, e.getMessage());
    }
  }
}