io.improbable.keanu.model.regression

Class RegressionModel<OUTPUT>

java.lang.Object

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

All Implemented Interfaces:

Model<DoubleTensor,OUTPUT>

public class RegressionModel<OUTPUT>
extends java.lang.Object
implements Model<DoubleTensor,OUTPUT>

A general linear regression model that can be fitted to input and output training data. You can construct a regression model as follows:

 RegressionModel model = RegressionModel
      .withTrainingData(inputTrainingData, outputTrainingData)
      .withRegularization(RegressionRegularization.RIDGE)
      .build();
 

Method Summary

Modifier and Type	Method and Description
double	getIntercept()
VertexId	getInterceptVertexId()
double	getWeight(int index)
DoubleTensor	getWeights()
VertexId	getWeightsVertexId()
OUTPUT	predict(DoubleTensor tensor)
static RegressionModelBuilder<BooleanTensor>	withTrainingData(DoubleTensor inputTrainingData, BooleanTensor outputTrainingData)
static RegressionModelBuilder<DoubleTensor>	withTrainingData(DoubleTensor inputTrainingData, DoubleTensor outputTrainingData)

Methods inherited from class java.lang.Object

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

Method Detail

withTrainingData

public static RegressionModelBuilder<DoubleTensor> withTrainingData(DoubleTensor inputTrainingData,
                                                                    DoubleTensor outputTrainingData)

withTrainingData

public static RegressionModelBuilder<BooleanTensor> withTrainingData(DoubleTensor inputTrainingData,
                                                                     BooleanTensor outputTrainingData)

getWeights

public DoubleTensor getWeights()

getIntercept

public double getIntercept()

getWeight

public double getWeight(int index)

predict

public OUTPUT predict(DoubleTensor tensor)

Specified by:

predict in interface Model<DoubleTensor,OUTPUT>

getInterceptVertexId

public VertexId getInterceptVertexId()

getWeightsVertexId

public VertexId getWeightsVertexId()