mlda.labelsDistribution.MeanEntropy.java Source code

Introduction

Here is the source code for mlda.labelsDistribution.MeanEntropy.java

Source

/*
 * This file is part of the MLDA.
 *
 * (c)  Jose Maria Moyano Murillo
 *      Eva Lucrecia Gibaja Galindo
 *      Sebastian Ventura Soto <sventura@uco.es>
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

package mlda.labelsDistribution;

import mlda.base.MLDataMetric;
import mlda.util.Utils;
import mulan.data.MultiLabelInstances;
import weka.core.AttributeStats;
import weka.core.Instances;

/**
* Class implementing the Mean of entropies of labels
*
* @author Jose Maria Moyano Murillo
*/
public class MeanEntropy extends MLDataMetric {

    /**
     * Constructor
     */
    public MeanEntropy() {
        super("Mean of entropies of labels");
    }

    /**
     * Calculate metric value
     * 
     * @param mlData Multi-label dataset to which calculate the metric
     * @return Value of the metric
     */
    public double calculate(MultiLabelInstances mlData) {
        Instances instances = mlData.getDataSet();

        int nLabels = mlData.getNumLabels();
        int[] labels = mlData.getLabelIndices();

        double[] entropies = new double[nLabels];

        for (int i = 0; i < nLabels; i++) {
            AttributeStats attStats = instances.attributeStats(labels[i]);

            if (attStats.nominalCounts != null) {
                entropies[i] = Utils.entropy(attStats.nominalCounts);
            }
        }

        double meanEntropy = 0;
        for (double e : entropies) {
            meanEntropy += e;
        }
        meanEntropy /= entropies.length;

        this.value = meanEntropy;

        return value;
    }

}