EnumeratedIntegerDistribution.java
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* https://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
/*
* This is not the original file distributed by the Apache Software Foundation
* It has been modified by the Hipparchus project
*/
package org.hipparchus.distribution.discrete;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import org.hipparchus.distribution.EnumeratedDistribution;
import org.hipparchus.exception.MathIllegalArgumentException;
import org.hipparchus.util.MathUtils;
import org.hipparchus.util.Pair;
/**
* Implementation of an integer-valued {@link EnumeratedDistribution}.
* <p>
* Values with zero-probability are allowed but they do not extend the
* support.
* <p>
* Duplicate values are allowed. Probabilities of duplicate values are combined
* when computing cumulative probabilities and statistics.
*/
public class EnumeratedIntegerDistribution extends AbstractIntegerDistribution {
/** Serializable UID. */
private static final long serialVersionUID = 20130308L;
/**
* {@link EnumeratedDistribution} instance (using the {@link Integer} wrapper)
* used to generate the pmf.
*/
private final EnumeratedDistribution<Integer> innerDistribution;
/**
* Create a discrete distribution using the given probability mass function
* definition.
*
* @param singletons array of random variable values.
* @param probabilities array of probabilities.
* @throws MathIllegalArgumentException if
* {@code singletons.length != probabilities.length}
* @throws MathIllegalArgumentException if probabilities contains negative, infinite or NaN values or only 0's
*/
public EnumeratedIntegerDistribution(final int[] singletons, final double[] probabilities)
throws MathIllegalArgumentException {
innerDistribution =
new EnumeratedDistribution<>(createDistribution(singletons, probabilities));
}
/**
* Create a discrete integer-valued distribution from the input data. Values are assigned
* mass based on their frequency. For example, [0,1,1,2] as input creates a distribution
* with values 0, 1 and 2 having probability masses 0.25, 0.5 and 0.25 respectively,
*
* @param data input dataset
*/
public EnumeratedIntegerDistribution(final int[] data) {
final Map<Integer, Integer> dataMap = new HashMap<>();
for (int value : data) {
Integer count = dataMap.get(value);
if (count == null) {
count = 0;
}
dataMap.put(value, ++count);
}
final int massPoints = dataMap.size();
final double denom = data.length;
final int[] values = new int[massPoints];
final double[] probabilities = new double[massPoints];
int index = 0;
for (Entry<Integer, Integer> entry : dataMap.entrySet()) {
values[index] = entry.getKey();
probabilities[index] = entry.getValue().intValue() / denom;
index++;
}
innerDistribution =
new EnumeratedDistribution<>(createDistribution(values, probabilities));
}
/**
* Create the list of Pairs representing the distribution from singletons and probabilities.
*
* @param singletons values
* @param probabilities probabilities
* @return list of value/probability pairs
* @throws MathIllegalArgumentException if probabilities contains negative, infinite or NaN values or only 0's
*/
private static List<Pair<Integer, Double>> createDistribution(int[] singletons,
double[] probabilities) {
MathUtils.checkDimension(singletons.length, probabilities.length);
final List<Pair<Integer, Double>> samples = new ArrayList<>(singletons.length);
final double[] normalizedProbabilities = EnumeratedDistribution.checkAndNormalize(probabilities);
for (int i = 0; i < singletons.length; i++) {
samples.add(new Pair<>(singletons[i], normalizedProbabilities[i]));
}
return samples;
}
/**
* {@inheritDoc}
*/
@Override
public double probability(final int x) {
return innerDistribution.probability(x);
}
/**
* {@inheritDoc}
*/
@Override
public double cumulativeProbability(final int x) {
double probability = 0;
for (final Pair<Integer, Double> sample : innerDistribution.getPmf()) {
if (sample.getKey() <= x) {
probability += sample.getValue();
}
}
return probability;
}
/**
* {@inheritDoc}
*
* @return {@code sum(singletons[i] * probabilities[i])}
*/
@Override
public double getNumericalMean() {
double mean = 0;
for (final Pair<Integer, Double> sample : innerDistribution.getPmf()) {
mean += sample.getValue() * sample.getKey();
}
return mean;
}
/**
* {@inheritDoc}
*
* @return {@code sum((singletons[i] - mean) ^ 2 * probabilities[i])}
*/
@Override
public double getNumericalVariance() {
double mean = 0;
double meanOfSquares = 0;
for (final Pair<Integer, Double> sample : innerDistribution.getPmf()) {
mean += sample.getValue() * sample.getKey();
meanOfSquares += sample.getValue() * sample.getKey() * sample.getKey();
}
return meanOfSquares - mean * mean;
}
/**
* {@inheritDoc}
*
* Returns the lowest value with non-zero probability.
*
* @return the lowest value with non-zero probability.
*/
@Override
public int getSupportLowerBound() {
int min = Integer.MAX_VALUE;
for (final Pair<Integer, Double> sample : innerDistribution.getPmf()) {
if (sample.getKey() < min && sample.getValue() > 0) {
min = sample.getKey();
}
}
return min;
}
/**
* {@inheritDoc}
*
* Returns the highest value with non-zero probability.
*
* @return the highest value with non-zero probability.
*/
@Override
public int getSupportUpperBound() {
int max = Integer.MIN_VALUE;
for (final Pair<Integer, Double> sample : innerDistribution.getPmf()) {
if (sample.getKey() > max && sample.getValue() > 0) {
max = sample.getKey();
}
}
return max;
}
/**
* {@inheritDoc}
*
* The support of this distribution is connected.
*
* @return {@code true}
*/
@Override
public boolean isSupportConnected() {
return true;
}
/**
* Return the probability mass function as a list of (value, probability) pairs.
*
* @return the probability mass function.
*/
public List<Pair<Integer, Double>> getPmf() {
return innerDistribution.getPmf();
}
}