io.improbable.keanu.network

Class KeanuProbabilisticModelWithGradient

java.lang.Object

io.improbable.keanu.network.KeanuProbabilisticModel

io.improbable.keanu.network.KeanuProbabilisticModelWithGradient

All Implemented Interfaces:

ProbabilisticModel, ProbabilisticModelWithGradient, java.lang.AutoCloseable

public class KeanuProbabilisticModelWithGradient
extends KeanuProbabilisticModel
implements ProbabilisticModelWithGradient

An implementation of ProbabilisticModelWithGradient that is backed by a BayesianNetwork

Constructor Summary

Constructors

Constructor and Description
KeanuProbabilisticModelWithGradient(BayesianNetwork bayesianNetwork)
KeanuProbabilisticModelWithGradient(java.util.Set<Vertex> variables)

Method Summary

Modifier and Type	Method and Description
java.util.Map<VariableReference,DoubleTensor>	logLikelihoodGradients()
java.util.Map<VariableReference,DoubleTensor>	logLikelihoodGradients(java.util.Map<VariableReference,?> inputs)
java.util.Map<VariableReference,DoubleTensor>	logProbGradients()
java.util.Map<VariableReference,DoubleTensor>	logProbGradients(java.util.Map<VariableReference,?> inputs)

Methods inherited from class io.improbable.keanu.network.KeanuProbabilisticModel

cascadeValues, getContinuousLatentVariables, getLatentOrObservedVertices, getLatentVariables, getLatentVertices, logLikelihood, logProb, logProbAfter

Methods inherited from class java.lang.Object

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

Methods inherited from interface io.improbable.keanu.algorithms.ProbabilisticModel

close, getContinuousLatentVariables, getLatentVariables, logLikelihood, logLikelihood, logProb, logProb, logProbAfter

Constructor Detail

KeanuProbabilisticModelWithGradient

public KeanuProbabilisticModelWithGradient(BayesianNetwork bayesianNetwork)

KeanuProbabilisticModelWithGradient

public KeanuProbabilisticModelWithGradient(java.util.Set<Vertex> variables)

Method Detail

logProbGradients

public java.util.Map<VariableReference,DoubleTensor> logProbGradients(java.util.Map<VariableReference,?> inputs)

Specified by:

logProbGradients in interface ProbabilisticModelWithGradient

logProbGradients

public java.util.Map<VariableReference,DoubleTensor> logProbGradients()

Specified by:

logProbGradients in interface ProbabilisticModelWithGradient

logLikelihoodGradients

public java.util.Map<VariableReference,DoubleTensor> logLikelihoodGradients(java.util.Map<VariableReference,?> inputs)

Specified by:

logLikelihoodGradients in interface ProbabilisticModelWithGradient

logLikelihoodGradients

public java.util.Map<VariableReference,DoubleTensor> logLikelihoodGradients()

Specified by:

logLikelihoodGradients in interface ProbabilisticModelWithGradient