org.deeplearning4j.models.rntn

Class RNTN.Builder

java.lang.Object

org.deeplearning4j.models.rntn.RNTN.Builder

Enclosing class:

RNTN

public static class RNTN.Builder
extends Object

Constructor Summary

Constructors

Constructor and Description
Builder()

Method Summary

Modifier and Type	Method and Description
RNTN	build()
RNTN.Builder	setActivationFunction(org.nd4j.linalg.api.activation.ActivationFunction activationFunction)
RNTN.Builder	setAdagradResetFrequency(int adagradResetFrequency)
RNTN.Builder	setBinaryClassificationSize(int binaryClassificationSize)
RNTN.Builder	setBinaryINd4jize(int binaryINd4jize)
RNTN.Builder	setBinaryTransformSize(int binaryTransformSize)
RNTN.Builder	setClassWeights(Map<Integer,Float> classWeights)
RNTN.Builder	setCombineClassification(boolean combineClassification)
RNTN.Builder	setFeatureVectors(Map<String,org.nd4j.linalg.api.ndarray.INDArray> featureVectors)
RNTN.Builder	setFeatureVectors(Word2Vec vec)
RNTN.Builder	setLowerCasefeatureNames(boolean lowerCasefeatureNames)
RNTN.Builder	setNumBinaryMatrices(int numBinaryMatrices)
RNTN.Builder	setNumHidden(int numHidden)
RNTN.Builder	setNumUnaryMatrices(int numUnaryMatrices)
RNTN.Builder	setRandomFeatureVectors(boolean randomFeatureVectors)
RNTN.Builder	setRegTransformINDArray(double regTransformINDArray)
RNTN.Builder	setRng(org.apache.commons.math3.random.RandomGenerator rng)
RNTN.Builder	setScalingForInit(double scalingForInit)
RNTN.Builder	setSimplifiedModel(boolean simplifiedModel)
RNTN.Builder	setUnaryClassificationSize(int unaryClassificationSize)
RNTN.Builder	setUseTensors(boolean useINd4j)
RNTN.Builder	withOutputActivation(org.nd4j.linalg.api.activation.ActivationFunction outputActivationFunction)

Methods inherited from class java.lang.Object

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

Constructor Detail

Builder

public Builder()

Method Detail

withOutputActivation

public RNTN.Builder withOutputActivation(org.nd4j.linalg.api.activation.ActivationFunction outputActivationFunction)

setFeatureVectors

public RNTN.Builder setFeatureVectors(Word2Vec vec)

setNumHidden

public RNTN.Builder setNumHidden(int numHidden)

setRng

public RNTN.Builder setRng(org.apache.commons.math3.random.RandomGenerator rng)

setUseTensors

public RNTN.Builder setUseTensors(boolean useINd4j)

setCombineClassification

public RNTN.Builder setCombineClassification(boolean combineClassification)

setSimplifiedModel

public RNTN.Builder setSimplifiedModel(boolean simplifiedModel)

setRandomFeatureVectors

public RNTN.Builder setRandomFeatureVectors(boolean randomFeatureVectors)

setScalingForInit

public RNTN.Builder setScalingForInit(double scalingForInit)

setLowerCasefeatureNames

public RNTN.Builder setLowerCasefeatureNames(boolean lowerCasefeatureNames)

setActivationFunction

public RNTN.Builder setActivationFunction(org.nd4j.linalg.api.activation.ActivationFunction activationFunction)

setAdagradResetFrequency

public RNTN.Builder setAdagradResetFrequency(int adagradResetFrequency)

setRegTransformINDArray

public RNTN.Builder setRegTransformINDArray(double regTransformINDArray)

setFeatureVectors

public RNTN.Builder setFeatureVectors(Map<String,org.nd4j.linalg.api.ndarray.INDArray> featureVectors)

setNumBinaryMatrices

public RNTN.Builder setNumBinaryMatrices(int numBinaryMatrices)

setBinaryTransformSize

public RNTN.Builder setBinaryTransformSize(int binaryTransformSize)

setBinaryINd4jize

public RNTN.Builder setBinaryINd4jize(int binaryINd4jize)

setBinaryClassificationSize

public RNTN.Builder setBinaryClassificationSize(int binaryClassificationSize)

setNumUnaryMatrices

public RNTN.Builder setNumUnaryMatrices(int numUnaryMatrices)

setUnaryClassificationSize

public RNTN.Builder setUnaryClassificationSize(int unaryClassificationSize)

setClassWeights

public RNTN.Builder setClassWeights(Map<Integer,Float> classWeights)

build

public RNTN build()