lu.lippmann.cdb.datasetview.tasks.HierarchizeReverseTask.java Source code

Introduction

Here is the source code for lu.lippmann.cdb.datasetview.tasks.HierarchizeReverseTask.java

Source

/**
 * Copyright 2014-2016 LIST (Luxembourg Institute of Science and Technology), all right reserved.
 * Authorship : Olivier PARISOT, Yoanne DIDRY
 * Licensed under GNU General Public License version 3
 */
package lu.lippmann.cdb.datasetview.tasks;

import org.apache.commons.lang.ArrayUtils;

import lu.lippmann.cdb.weka.*;
import weka.core.Instances;

/**
 * Task.
 * 
 * @author the WP1 team
 */
public final class HierarchizeReverseTask extends Task {
    /**
     * {@inheritDoc}
     */
    @Override
    String getName() {
        return "Hierarchize (reverse)";
    }

    /**
     * {@inheritDoc}
     */
    @Override
    String getIconPath() {
        return "menu/hierarchize.png";
    }

    /**
     * {@inheritDoc}
     */
    @Override
    Instances process0(final Instances dataSet) throws Exception {
        if (dataSet.classIndex() == -1)
            throw new Exception("Need a selected class!");
        final int[] idx = WekaMachineLearningUtil.computeRankedAttributes(dataSet);
        ArrayUtils.reverse(idx);
        return WekaDataProcessingUtil.buildFilteredByAttributesDataSet(dataSet, idx);
    }

}