TestData.java

package LogisticRegression;

import java.io.*;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Map;

/**
 * Created by jay on 11/29/15.
 */
public class TestData {

    private String pathToFile;
    public Map<String, Integer> testLabelCount;

    public TestData(String pathToFile) {
        this.pathToFile = pathToFile;
        this.testLabelCount = new HashMap<>();
    }


    public ArrayList<Example> testData() {

        ArrayList<Example> test = new ArrayList<Example>();
        BufferedReader br;


        try {
            br = new BufferedReader(new FileReader(pathToFile));
            String line;
            String ignoreLine = br.readLine();
            int index = 0;

            while ((line = br.readLine()) != null) {
                String perLine[] = line.split(",");

                Example testExample = new Example();
                testExample.setIndex(index);

                double[] values = new double[perLine.length - 1];

                for (int i = 0; i < perLine.length; i++) {
                    if (i == perLine.length - 1) {
                        testExample.setActualLabel(perLine[i]);
                        addInMap(perLine[i]);
                    } else {
                        values[i] = Double.parseDouble(perLine[i]);
                    }
                }

                index++;
                testExample.setValues(values);
                test.add(testExample);
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
        return test;
    }

    private void addInMap(String label) {
        if (testLabelCount.containsKey(label)) {
            Integer count = testLabelCount.get(label);
            testLabelCount.put(label, ++count);
        } else {
            testLabelCount.put(label, 1);
        }
    }
}