com.github.TKnudsen.ComplexDataObject.model.io.arff.ARFFInstancesIO Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of complex-data-object Show documentation

A library that models real-world objects in Java, referred to as ComplexDataObjects. Other features: IO and preprocessing of ComplexDataObjects.

The newest version!

package com.github.TKnudsen.ComplexDataObject.model.io.arff;

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

import weka.core.Instances;
import weka.core.converters.ArffSaver;
import weka.core.converters.ConverterUtils.DataSink;

/**
 * 
 * Title: ARFFInstancesIO
 * 
 * 
 * 
 * Description: Tools for the IO if Weka Instanes to *.arff and vice versa.
 * 
 * 
 * 
 * Copyright: Copyright (c) 2018-2019
 * 
 * 
 * @author Juergen Bernard
 * @version 1.02
 */
public class ARFFInstancesIO {

	public static Instances loadARFF(String arffFile) {
		if (arffFile == null)
			return null;

		BufferedReader reader;
		try {
			FileReader fileReader = new FileReader(arffFile);
			reader = new BufferedReader(fileReader);

			Instances instances = new Instances(reader);

			fileReader.close();
			reader.close();

			return instances;
		} catch (FileNotFoundException e) {
			e.printStackTrace();
			return null;
		} catch (IOException e) {
			System.err.println("ARFFInstancesIO.loadARFF: unable to read file " + arffFile + " . Exception caught.");
			return null;
		}
	}

	/**
	 * simple Weka writer routine. Inspired by Weka's tutorial:
	 * https://weka.wikispaces.com/Save+Instances+to+an+ARFF+File
	 * 
	 * @param instances
	 * @param fileNameWithArffExtension
	 * @throws IOException
	 */
	public static void saveARFF(Instances instances, String fileNameWithArffExtension) throws IOException {
		try {
			DataSink.write(fileNameWithArffExtension, instances);
		} catch (Exception e) {
			System.err.println("Failed to save data to: " + fileNameWithArffExtension);
			e.printStackTrace();
		}
	}

	public static void saveARFF(Instances instances, File arffFile) throws IOException {
		ArffSaver saver = new ArffSaver();
		saver.setInstances(instances);
		saver.setFile(arffFile);
		saver.writeBatch();
	}
}