io.improbable.keanu.distributions.discrete

Class Categorical<CATEGORY,TENSOR extends Tensor<CATEGORY>>

java.lang.Object

io.improbable.keanu.distributions.discrete.Categorical<CATEGORY,TENSOR>

All Implemented Interfaces:

Distribution<TENSOR>

public class Categorical<CATEGORY,TENSOR extends Tensor<CATEGORY>>
extends java.lang.Object
implements Distribution<TENSOR>

Method Summary

Modifier and Type	Method and Description
DoubleTensor	logProb(TENSOR x)
TENSOR	sample(long[] shape, KeanuRandom random)
static <CAT,TENSOR extends Tensor<CAT>> Categorical<CAT,TENSOR>	withParameters(java.util.Map<CAT,DoubleTensor> selectableValues)

Methods inherited from class java.lang.Object

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

Method Detail

withParameters

public static <CAT,TENSOR extends Tensor<CAT>> Categorical<CAT,TENSOR> withParameters(java.util.Map<CAT,DoubleTensor> selectableValues)

sample

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

Specified by:

sample in interface Distribution<TENSOR extends Tensor<CATEGORY>>

logProb

public DoubleTensor logProb(TENSOR x)

Specified by:

logProb in interface Distribution<TENSOR extends Tensor<CATEGORY>>