org.deeplearning4j.spark.models.sequencevectors

Class SparkSequenceVectors.Builder<T extends SequenceElement>

java.lang.Object

org.deeplearning4j.spark.models.sequencevectors.SparkSequenceVectors.Builder<T>

Direct Known Subclasses:

SparkWord2Vec.Builder

Enclosing class:

SparkSequenceVectors<T extends SequenceElement>

public static class SparkSequenceVectors.Builder<T extends SequenceElement>
extends java.lang.Object

Field Summary

Fields

Modifier and Type	Field and Description
protected VectorsConfiguration	configuration
protected SparkModelExporter<T>	modelExporter
protected org.nd4j.parameterserver.distributed.conf.VoidConfiguration	peersConfiguration
protected org.apache.spark.storage.StorageLevel	storageLevel
protected int	workers

Constructor Summary

Constructors

Constructor and Description
Builder() Deprecated.
Builder(org.nd4j.parameterserver.distributed.conf.VoidConfiguration psConfiguration)
Builder(org.nd4j.parameterserver.distributed.conf.VoidConfiguration psConfiguration, VectorsConfiguration w2vConfiguration)

Method Summary

Modifier and Type	Method and Description
SparkSequenceVectors<T>	build()
SparkSequenceVectors.Builder<T>	epochs(int num)
SparkSequenceVectors.Builder<T>	iterations(int num)
SparkSequenceVectors.Builder<T>	layerSize(int layerSize)
SparkSequenceVectors.Builder<T>	minWordFrequency(int num)
SparkSequenceVectors.Builder<T>	negativeSampling(long samples)
SparkSequenceVectors.Builder<T>	setElementsLearningAlgorithm(SparkElementsLearningAlgorithm ela)
SparkSequenceVectors.Builder<T>	setLearningRate(double lr)
SparkSequenceVectors.Builder<T>	setModelExporter(SparkModelExporter<T> modelExporter)
SparkSequenceVectors.Builder<T>	setParameterServerConfiguration(org.nd4j.parameterserver.distributed.conf.VoidConfiguration configuration)
SparkSequenceVectors.Builder<T>	setSequenceLearningAlgorithm(SparkSequenceLearningAlgorithm sla)
SparkSequenceVectors.Builder<T>	setStorageLevel(org.apache.spark.storage.StorageLevel level)
SparkSequenceVectors.Builder<T>	subsampling(double rate)
SparkSequenceVectors.Builder<T>	useHierarchicSoftmax(boolean reallyUse)
SparkSequenceVectors.Builder<T>	workers(int num)

Methods inherited from class java.lang.Object

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

Field Detail

configuration

protected VectorsConfiguration configuration

modelExporter

protected SparkModelExporter<T extends SequenceElement> modelExporter

peersConfiguration

protected org.nd4j.parameterserver.distributed.conf.VoidConfiguration peersConfiguration

workers

protected int workers

storageLevel

protected org.apache.spark.storage.StorageLevel storageLevel

Constructor Detail

Builder

@Deprecated
public Builder()

Deprecated.

This method should NOT be used in real world environment

Builder

public Builder(@NonNull
               org.nd4j.parameterserver.distributed.conf.VoidConfiguration psConfiguration)

Builder

public Builder(@NonNull
               org.nd4j.parameterserver.distributed.conf.VoidConfiguration psConfiguration,
               @NonNull
               VectorsConfiguration w2vConfiguration)

Method Detail

setStorageLevel

public SparkSequenceVectors.Builder<T> setStorageLevel(org.apache.spark.storage.StorageLevel level)

Parameters:

level -

Returns:

minWordFrequency

public SparkSequenceVectors.Builder<T> minWordFrequency(int num)

Parameters:

num -

Returns:

workers

public SparkSequenceVectors.Builder<T> workers(int num)

Parameters:

num -

Returns:

setLearningRate

public SparkSequenceVectors.Builder<T> setLearningRate(double lr)

Parameters:

lr -

Returns:

setParameterServerConfiguration

public SparkSequenceVectors.Builder<T> setParameterServerConfiguration(@NonNull
                                                                       org.nd4j.parameterserver.distributed.conf.VoidConfiguration configuration)

Parameters:

configuration -

Returns:

setModelExporter

public SparkSequenceVectors.Builder<T> setModelExporter(@NonNull
                                                        SparkModelExporter<T> modelExporter)

Parameters:

modelExporter -

Returns:

epochs

public SparkSequenceVectors.Builder<T> epochs(int num)

Parameters:

num -

Returns:

iterations

public SparkSequenceVectors.Builder<T> iterations(int num)

Parameters:

num -

Returns:

subsampling

public SparkSequenceVectors.Builder<T> subsampling(double rate)

Parameters:

rate -

Returns:

useHierarchicSoftmax

public SparkSequenceVectors.Builder<T> useHierarchicSoftmax(boolean reallyUse)

Parameters:

reallyUse -

Returns:

negativeSampling

public SparkSequenceVectors.Builder<T> negativeSampling(long samples)

Parameters:

samples -

Returns:

setElementsLearningAlgorithm

public SparkSequenceVectors.Builder<T> setElementsLearningAlgorithm(@NonNull
                                                                    SparkElementsLearningAlgorithm ela)

Parameters:

ela -

Returns:

setSequenceLearningAlgorithm

public SparkSequenceVectors.Builder<T> setSequenceLearningAlgorithm(@NonNull
                                                                    SparkSequenceLearningAlgorithm sla)

Parameters:

sla -

Returns:

layerSize

public SparkSequenceVectors.Builder<T> layerSize(int layerSize)

build

public SparkSequenceVectors<T> build()