Class CBOWNetwork.Builder

java.lang.Object

deepnetts.net.CBOWNetwork.Builder

Enclosing class:

CBOWNetwork

public static class CBOWNetwork.Builder
extends Object

Constructor Summary

Constructors

Constructor	Description
Builder()

Method Summary

Modifier and Type	Method	Description
CBOWNetwork.Builder	addCBOWEmbeddingLayer(int vocabSize, int embeddingDim)
CBOWNetwork.Builder	addFullyConnectedLayer(int layerWidth, ActivationType activationType)
CBOWNetwork.Builder	addInputLayer(int layerWidth)
CBOWNetwork.Builder	addInputLayer(int layerWidth, int batchSize)
CBOWNetwork.Builder	addOutputLayer(int width, ActivationType activationType)
CBOWNetwork	build()
CBOWNetwork.Builder	lossFunction(LossType lossType)
CBOWNetwork.Builder	randomSeed(long seed)

Methods inherited from class Object

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

Constructor Details

Builder

public Builder()

Method Details

addInputLayer

public CBOWNetwork.Builder addInputLayer(int layerWidth)

addInputLayer

public CBOWNetwork.Builder addInputLayer(int layerWidth,
 int batchSize)

addCBOWEmbeddingLayer

public CBOWNetwork.Builder addCBOWEmbeddingLayer(int vocabSize,
 int embeddingDim)

addFullyConnectedLayer

public CBOWNetwork.Builder addFullyConnectedLayer(int layerWidth,
 ActivationType activationType)

addOutputLayer

public CBOWNetwork.Builder addOutputLayer(int width,
 ActivationType activationType)

lossFunction

public CBOWNetwork.Builder lossFunction(LossType lossType)

randomSeed

public CBOWNetwork.Builder randomSeed(long seed)

build

public CBOWNetwork build()