BoltzmannDistribution.java 文件源码

/**
 * Samples the output probability distribution.
 * @return the index of the sampled element
 */
public int sample(){

    if(this.needsUpdate){
        this.computeProbs();
    }

    double r = this.rand.nextDouble();
    double sum = 0.;
    for(int i = 0; i < this.probs.length; i++){
        sum += this.probs[i];
        if(r < sum){
            return i;
        }
    }

    throw new RuntimeErrorException(new Error("Error in sample; Boltzmann distribution did not sum to 1"));

}