Package io.github.kirstenali.deepj.transformer

Class DeepJOrbitTransformerBuilder

java.lang.Object

io.github.kirstenali.deepj.transformer.DeepJOrbitTransformerBuilder

public final class DeepJOrbitTransformerBuilder
extends Object

Constructor Summary

Constructors

Constructor	Description
DeepJOrbitTransformerBuilder()

Method Summary

Modifier and Type	Method	Description
TransformerStack	build()
DeepJOrbitTransformerBuilder	dFF(int dFF)
DeepJOrbitTransformerBuilder	dModel(int dModel)
DeepJOrbitTransformerBuilder	maxSeqLen(int maxSeqLen)
DeepJOrbitTransformerBuilder	nHeads(int nHeads)
DeepJOrbitTransformerBuilder	nLayers(int nLayers)
DeepJOrbitTransformerBuilder	parameters(ParameterFactory factory)
DeepJOrbitTransformerBuilder	random(Random rnd)
DeepJOrbitTransformerBuilder	seed(long seed)

Methods inherited from class java.lang.Object

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

Constructor Details

DeepJOrbitTransformerBuilder

public DeepJOrbitTransformerBuilder()

Method Details

parameters

public DeepJOrbitTransformerBuilder parameters(ParameterFactory factory)

dModel

public DeepJOrbitTransformerBuilder dModel(int dModel)

nHeads

public DeepJOrbitTransformerBuilder nHeads(int nHeads)

dFF

public DeepJOrbitTransformerBuilder dFF(int dFF)

nLayers

public DeepJOrbitTransformerBuilder nLayers(int nLayers)

maxSeqLen

public DeepJOrbitTransformerBuilder maxSeqLen(int maxSeqLen)

seed

public DeepJOrbitTransformerBuilder seed(long seed)

random

public DeepJOrbitTransformerBuilder random(Random rnd)

build

public TransformerStack build()