Package org.deeplearning4j.models.sequencevectors.transformers.impl

Class GraphTransformer.Builder<T extends SequenceElement>

java.lang.Object

org.deeplearning4j.models.sequencevectors.transformers.impl.GraphTransformer.Builder<T>

Enclosing class:

GraphTransformer<T extends SequenceElement>

public static class GraphTransformer.Builder<T extends SequenceElement>
extends Object

Field Summary

Fields

Modifier and Type	Field	Description
protected LabelsProvider<T>	labelsProvider
protected boolean	shuffle
protected IGraph<T,?>	sourceGraph
protected VocabCache<T>	vocabCache
protected GraphWalker<T>	walker

Constructor Summary

Constructors

Constructor	Description
Builder()
Builder(@NonNull IGraph<T,?> sourceGraph)
Builder(@NonNull GraphWalker<T> walker)

Method Summary

Modifier and Type	Method	Description
GraphTransformer<T>	build()
GraphTransformer.Builder<T>	setGraphWalker(@NonNull GraphWalker<T> walker)
GraphTransformer.Builder<T>	setLabelsProvider(@NonNull LabelsProvider<T> provider)
GraphTransformer.Builder<T>	setVocabCache(@NonNull VocabCache<T> vocabCache)
GraphTransformer.Builder<T>	shuffleOnReset(boolean reallyShuffle)

Methods inherited from class java.lang.Object

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

Field Detail

sourceGraph

protected IGraph<T extends SequenceElement,?> sourceGraph

labelsProvider

protected LabelsProvider<T extends SequenceElement> labelsProvider

walker

protected GraphWalker<T extends SequenceElement> walker

shuffle

protected boolean shuffle

vocabCache

protected VocabCache<T extends SequenceElement> vocabCache

Constructor Detail

Builder

public Builder()

Builder

public Builder(@NonNull
               @NonNull GraphWalker<T> walker)

Builder

public Builder(@NonNull
               @NonNull IGraph<T,?> sourceGraph)

Method Detail

setLabelsProvider

public GraphTransformer.Builder<T> setLabelsProvider(@NonNull
                                                     @NonNull LabelsProvider<T> provider)

setGraphWalker

public GraphTransformer.Builder<T> setGraphWalker(@NonNull
                                                  @NonNull GraphWalker<T> walker)

setVocabCache

public GraphTransformer.Builder<T> setVocabCache(@NonNull
                                                 @NonNull VocabCache<T> vocabCache)

shuffleOnReset

public GraphTransformer.Builder<T> shuffleOnReset(boolean reallyShuffle)

build

public GraphTransformer<T> build()