org.deeplearning4j.nn

Class LogisticRegression.Builder

java.lang.Object

org.deeplearning4j.nn.LogisticRegression.Builder

Enclosing class:

LogisticRegression

public static class LogisticRegression.Builder
extends Object

Constructor Summary

Constructors

Constructor and Description
LogisticRegression.Builder()

Method Summary

Methods

Modifier and Type	Method and Description
LogisticRegression	build()
LogisticRegression.Builder	numberOfInputs(int nIn)
LogisticRegression.Builder	numberOfOutputs(int nOut)
LogisticRegression.Builder	useRegularization(boolean regularize)
LogisticRegression.Builder	withBias(org.jblas.DoubleMatrix b)
LogisticRegression.Builder	withL2(double l2)
LogisticRegression.Builder	withWeights(org.jblas.DoubleMatrix W)

Methods inherited from class java.lang.Object

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

Constructor Detail

LogisticRegression.Builder

public LogisticRegression.Builder()

Method Detail

withL2

public LogisticRegression.Builder withL2(double l2)

useRegularization

public LogisticRegression.Builder useRegularization(boolean regularize)

withWeights

public LogisticRegression.Builder withWeights(org.jblas.DoubleMatrix W)

withBias

public LogisticRegression.Builder withBias(org.jblas.DoubleMatrix b)

numberOfInputs

public LogisticRegression.Builder numberOfInputs(int nIn)

numberOfOutputs

public LogisticRegression.Builder numberOfOutputs(int nOut)

build

public LogisticRegression build()