activeSegmentation.learning.WekaDataSet.java Source code

Introduction

Here is the source code for activeSegmentation.learning.WekaDataSet.java

Source

/*
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
package activeSegmentation.learning;

import java.io.FileReader;
import java.io.IOException;
import java.util.logging.Level;
import java.util.logging.Logger;

import activeSegmentation.IDataSet;
import weka.core.Instance;
import weka.core.Instances;

public class WekaDataSet implements IDataSet {

    private Instances dataset;

    /**
     * Constructs a WekaDataset
     *
     * @param arffFilePath The path to the arff file
     */
    public WekaDataSet(String arffFilePath) {
        try {
            dataset = new Instances(new FileReader(arffFilePath));
        } catch (IOException ex) {
            Logger.getLogger(WekaDataSet.class.getName()).log(Level.SEVERE, null, ex);
        }

    }

    /**
     * Creates a Weka dataset from an Instances object
     *
     * @param dataset The weka dataset
     */
    public WekaDataSet(Instances dataset) {
        this.dataset = new Instances(dataset);
    }

    /**
     * Creates a Weka Dataset from other Weka Dataset.
     *
     * @param dataset The dataset to use.
     */
    public WekaDataSet(IDataSet dataset) {
        this.dataset = new Instances(dataset.getDataset());
    }

    /**
     * Creates a Weka Dataset from a portion of the IDataset object
     *
     * @param dataset The dataset
     * @param first The position of the first instance to copy
     * @param toCopy The number of instances to copy
     */
    public WekaDataSet(IDataSet dataset, int first, int toCopy) {
        this.dataset = new Instances(dataset.getDataset(), first, toCopy);
    }

    @Override
    public int getNumAttributes() {
        return dataset.numAttributes();
    }

    /**
     * Set the index of the class attribute
     *
     * @param classIndex The index of the class attribute
     */
    public void setClassIndex(int classIndex) {
        dataset.setClassIndex(classIndex);
    }

    @Override
    public IDataSet copy() {
        return new WekaDataSet(this);
    }

    @Override
    public void addAll(IDataSet dataset) {
        this.dataset.addAll(dataset.getDataset());
    }

    @Override
    public int getNumInstances() {
        return dataset.numInstances();
    }

    @Override
    public Instance instance(int index) {
        return dataset.instance(index);
    }

    @Override
    public void set(int index, Instance instance) {
        dataset.set(index, instance);
    }

    public int getNumClasses() {
        return dataset.numClasses();
    }

    public int getClassIndex() {
        return dataset.classIndex();
    }

    @Override
    public void add(Instance instance) {
        dataset.add(instance);
    }

    @Override
    public void remove(int index) {
        dataset.remove(index);
    }

    @Override
    public Instances getDataset() {
        return dataset;
    }

    @Override
    public void delete() {
        dataset.delete();
        dataset = null;
    }

    @Override
    public boolean isEmpty() {
        // TODO Auto-generated method stub
        return false;
    }

}