io.improbable.keanu.model.regression

Class LinearRegressionGraph<OUTPUT>

java.lang.Object

io.improbable.keanu.model.regression.LinearRegressionGraph<OUTPUT>

All Implemented Interfaces:

ModelGraph<DoubleTensor,OUTPUT>

public class LinearRegressionGraph<OUTPUT>
extends java.lang.Object
implements ModelGraph<DoubleTensor,OUTPUT>

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	LinearRegressionGraph.OutputVertices<OUTPUT>

Constructor Summary

Constructors

Constructor and Description
LinearRegressionGraph(long[] featureShape, java.util.function.Function<DoubleVertex,LinearRegressionGraph.OutputVertices<OUTPUT>> outputTransform, DoubleVertex interceptVertex, DoubleVertex weightsVertex)

Method Summary

Modifier and Type	Method and Description
DoubleVertex	getInterceptVertex()
Vertex<OUTPUT>	getOutputVertex()
DoubleVertex	getWeightVertex()
void	observeValues(DoubleTensor input, OUTPUT output)
OUTPUT	predict(DoubleTensor input)

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.model.ModelGraph

getBayesianNetwork

Constructor Detail

LinearRegressionGraph

public LinearRegressionGraph(long[] featureShape,
                             java.util.function.Function<DoubleVertex,LinearRegressionGraph.OutputVertices<OUTPUT>> outputTransform,
                             DoubleVertex interceptVertex,
                             DoubleVertex weightsVertex)

Method Detail

predict

public OUTPUT predict(DoubleTensor input)

observeValues

public void observeValues(DoubleTensor input,
                          OUTPUT output)

Specified by:

observeValues in interface ModelGraph<DoubleTensor,OUTPUT>

getInterceptVertex

public DoubleVertex getInterceptVertex()

getWeightVertex

public DoubleVertex getWeightVertex()

getOutputVertex

public Vertex<OUTPUT> getOutputVertex()