Main.java

 public static void main(String[] args) throws IOException, InterruptedException {

        int numLinesToSkip = 0;
        String delimiter = ",";
        RecordReader recordReader = new CSVRecordReader(numLinesToSkip, delimiter);
        recordReader.initialize(new FileSplit(new org.springframework.core.io.ClassPathResource("g.csv").getFile()));
        int labelIndex = 5;     // 1 based
        int batchSize = 50;
        DataSetIterator iterator = new RecordReaderDataSetIterator(recordReader, batchSize, labelIndex);
        org.nd4j.linalg.dataset.api.DataSet next = iterator.next();
        SplitTestAndTrain testAndTrain = next.splitTestAndTrain(0.8);  //Use 80% of data for training
        //...
}