examples.stats.example2.Example2 Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jstat Show documentation
Java Library for Statistical Analysis.
The newest version!
package examples.stats.example2;

import dataloader.CsvDataLoader;
import datasets.VectorDouble;
import datastructs.IVector;
import org.apache.commons.math3.stat.inference.TestUtils;
import utils.ListUtils;

import java.io.File;
import java.io.IOException;
import java.util.List;
import java.util.Map;

/**
 *
 * Category: Statistics, Hypothesis Testing
 * ID: HypothesisTestingForMean
 * Description: Load a CSV file using the CsvDataLoader.MapLoader and create
 * a NumericSample from the data set. Use the sample for hypothesis testing
 */

public class Example2 {


    public static void main(String[] args){

        try {
            // load data set
            Map> dataSet = CsvDataLoader.MapLoader.parseFile(new File("data/robot_state.csv"));
            IVector sample = CsvDataLoader.MapLoader.buildNumericSample(dataSet, "X");

            // the mean value we assume
            double mu = 2.85;
            double level = 0.05;
            boolean rejectH0 = TestUtils.tTest(mu, ListUtils.toDoubleArray(sample.toArray()), level );
            double pLevel = TestUtils.tTest(mu, ListUtils.toDoubleArray(sample.toArray()));

            System.out.println("p-level is: "+pLevel);
            if(rejectH0){
                System.out.println("H0 hypothesis: mu="+mu+" can be rejected with confidence: "+(1.0-level));
            }
            else{
                System.out.println("H0 hypothesis: mu="+mu+" can not be rejected");
            }

            // now we should not reject
            mu = ((VectorDouble)sample).getMean();

            rejectH0 = TestUtils.tTest(mu, ListUtils.toDoubleArray(sample.toArray()), level );
            pLevel = TestUtils.tTest(mu, ListUtils.toDoubleArray(sample.toArray()));

            System.out.println("p-level is: "+pLevel);
            if(rejectH0){
                System.out.println("H0 hypothesis: mu="+mu+" can be rejected with confidence: "+(1.0-level));
            }
            else{
                System.out.println("H0 hypothesis: mu="+mu+" can not be rejected");
            }
        }
        catch(IOException exception)
        {
            System.out.println(exception.toString());
        }


    }
}