io.improbable.keanu.distributions.discrete

Class Geometric

java.lang.Object

io.improbable.keanu.distributions.discrete.Geometric

All Implemented Interfaces:

DiscreteDistribution, Distribution<IntegerTensor>

public class Geometric
extends java.lang.Object
implements DiscreteDistribution

Implements a Geometric Random Distribution. More details can be found at: https://en.wikipedia.org/wiki/Geometric_distribution

Method Summary

Modifier and Type	Method and Description
DoubleTensor	logProb(IntegerTensor k)
static DoubleVertex	logProbOutput(IntegerPlaceholderVertex k, DoublePlaceholderVertex p)
IntegerTensor	sample(long[] shape, KeanuRandom random)
static DiscreteDistribution	withParameters(DoubleTensor p)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Detail

withParameters

public static DiscreteDistribution withParameters(DoubleTensor p)

sample

public IntegerTensor sample(long[] shape,
                            KeanuRandom random)

Specified by:

sample in interface Distribution<IntegerTensor>

logProb

public DoubleTensor logProb(IntegerTensor k)

Specified by:

logProb in interface Distribution<IntegerTensor>

logProbOutput

public static DoubleVertex logProbOutput(IntegerPlaceholderVertex k,
                                         DoublePlaceholderVertex p)