Class SkipGramNetwork.Builder

java.lang.Object

deepnetts.net.SkipGramNetwork.Builder

Enclosing class:

SkipGramNetwork

public static class SkipGramNetwork.Builder
extends Object

Constructor Summary

Constructors

Constructor	Description
Builder()

Method Summary

Modifier and Type	Method	Description
SkipGramNetwork.Builder	addFullyConnectedLayer(int layerWidth, ActivationType activationType)
SkipGramNetwork.Builder	addInputLayer(int layerWidth)
SkipGramNetwork.Builder	addInputLayer(int layerWidth, int batchSize)
SkipGramNetwork.Builder	addOutputLayer(int width, ActivationType activationType)
SkipGramNetwork.Builder	addSkipGramEmbeddingLayer(int vocabSize, int embeddingDim)
SkipGramNetwork	build()
SkipGramNetwork.Builder	lossFunction(LossType lossType)
SkipGramNetwork.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 SkipGramNetwork.Builder addInputLayer(int layerWidth)

addInputLayer

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

addSkipGramEmbeddingLayer

public SkipGramNetwork.Builder addSkipGramEmbeddingLayer(int vocabSize,
 int embeddingDim)

addFullyConnectedLayer

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

addOutputLayer

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

lossFunction

public SkipGramNetwork.Builder lossFunction(LossType lossType)

randomSeed

public SkipGramNetwork.Builder randomSeed(long seed)

build

public SkipGramNetwork build()