Package io.github.kirstenali.deepj.transformer

Class DeepJOriginTransformerBuilder

java.lang.Object

io.github.kirstenali.deepj.transformer.DeepJOriginTransformerBuilder

public final class DeepJOriginTransformerBuilder
extends Object

Constructor Summary

Constructors

Constructor	Description
DeepJOriginTransformerBuilder()

Method Summary

Modifier and Type	Method	Description
TransformerStack	build()
DeepJOriginTransformerBuilder	dFF(int dFF)
DeepJOriginTransformerBuilder	dModel(int dModel)
DeepJOriginTransformerBuilder	ffnActivation(Supplier<ActivationFunction> factory)
DeepJOriginTransformerBuilder	nHeads(int nHeads)
DeepJOriginTransformerBuilder	nLayers(int nLayers)
DeepJOriginTransformerBuilder	parameters(ParameterFactory factory)
DeepJOriginTransformerBuilder	random(Random rnd)
DeepJOriginTransformerBuilder	seed(long seed)

Methods inherited from class java.lang.Object

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

Constructor Details

DeepJOriginTransformerBuilder

public DeepJOriginTransformerBuilder()

Method Details

parameters

public DeepJOriginTransformerBuilder parameters(ParameterFactory factory)

dModel

public DeepJOriginTransformerBuilder dModel(int dModel)

nHeads

public DeepJOriginTransformerBuilder nHeads(int nHeads)

dFF

public DeepJOriginTransformerBuilder dFF(int dFF)

nLayers

public DeepJOriginTransformerBuilder nLayers(int nLayers)

ffnActivation

public DeepJOriginTransformerBuilder ffnActivation(Supplier<ActivationFunction> factory)

seed

public DeepJOriginTransformerBuilder seed(long seed)

random

public DeepJOriginTransformerBuilder random(Random rnd)

build

public TransformerStack build()