org.nd4j.linalg.api.ops.aggregates.impl

Class AggregateSkipGram

java.lang.Object

org.nd4j.linalg.api.ops.aggregates.BaseAggregate

org.nd4j.linalg.api.ops.aggregates.impl.AggregateSkipGram

All Implemented Interfaces:

Aggregate

public class AggregateSkipGram
extends BaseAggregate

This aggregate encapsulates AggregateSkipGram training round for a given word and context

Author:

[email protected]

Field Summary

Fields inherited from class org.nd4j.linalg.api.ops.aggregates.BaseAggregate

arguments, finalResult, indexingArguments, intArrayArguments, realArguments, shapes

Constructor Summary

Constructors

Constructor and Description
AggregateSkipGram(INDArray syn0, INDArray syn1, INDArray syn1Neg, INDArray expTable, INDArray negTable, int idxSyn0, int[] idxSyn1, int[] codes, int negativeRounds, int ngStarter, int vectorLength, double alpha, long nextRandom, int vocabSize)
AggregateSkipGram(INDArray syn0, INDArray syn1, INDArray syn1Neg, INDArray expTable, INDArray negTable, int idxSyn0, int[] idxSyn1, int[] codes, int negativeRounds, int ngStarter, int vectorLength, double alpha, long nextRandom, int vocabSize, INDArray inferenceVector)
AggregateSkipGram(int w1, int w2, int[] codes, int[] points, int negSamples, double lr, int vectorLength) This is special signature suitable for use with VoidParameterServer, never ever use it outside of spark-nlp

Method Summary

Modifier and Type	Method and Description
int	getSharedMemorySize() This method returns amount of shared memory required for this specific Aggregate.
int	getThreadsPerInstance() This method returns desired number of threads per Aggregate instance PLEASE NOTE: this method is especially important for CUDA backend.
int	maxArguments() This method returns maximum number of shapes being passed per Aggregate
int	maxIndexArguments() This method returns maximum number of IndexArguments per Aggregate
int	maxIntArrays() This method returns maximum number of IntArrays being passed per Aggregate
int	maxIntArraySize() This method returns maximum length for IntArrays, if any
int	maxRealArguments() This method returns maximum number of real (float/double) per Aggregate
int	maxShapes() This method returns maximum number of shapes being passed per Aggregate
String	name()
int	opNum()

Methods inherited from class org.nd4j.linalg.api.ops.aggregates.BaseAggregate

getArguments, getFinalResult, getIndexingArguments, getIntArrayArguments, getRealArguments, getRequiredBatchMemorySize, getShapes, setFinalResult

Methods inherited from class java.lang.Object

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

Constructor Detail

AggregateSkipGram

public AggregateSkipGram(INDArray syn0,
                         INDArray syn1,
                         INDArray syn1Neg,
                         INDArray expTable,
                         INDArray negTable,
                         int idxSyn0,
                         int[] idxSyn1,
                         int[] codes,
                         int negativeRounds,
                         int ngStarter,
                         int vectorLength,
                         double alpha,
                         long nextRandom,
                         int vocabSize,
                         INDArray inferenceVector)

AggregateSkipGram

public AggregateSkipGram(@NonNull
                         INDArray syn0,
                         INDArray syn1,
                         INDArray syn1Neg,
                         @NonNull
                         INDArray expTable,
                         INDArray negTable,
                         int idxSyn0,
                         int[] idxSyn1,
                         int[] codes,
                         int negativeRounds,
                         int ngStarter,
                         int vectorLength,
                         double alpha,
                         long nextRandom,
                         int vocabSize)

AggregateSkipGram

public AggregateSkipGram(int w1,
                         int w2,
                         int[] codes,
                         int[] points,
                         int negSamples,
                         double lr,
                         int vectorLength)

This is special signature suitable for use with VoidParameterServer, never ever use it outside of spark-nlp

Parameters:

w1 -

w2 -

lr -

vectorLength -

Method Detail

getSharedMemorySize

public int getSharedMemorySize()

This method returns amount of shared memory required for this specific Aggregate. PLEASE NOTE: this method is especially important for CUDA backend. On CPU backend it might be ignored, depending on Aggregate.

Returns:

getThreadsPerInstance

public int getThreadsPerInstance()

This method returns desired number of threads per Aggregate instance PLEASE NOTE: this method is especially important for CUDA backend. On CPU backend it might be ignored, depending on Aggregate.

Returns:

name

public String name()

Returns:

opNum

public int opNum()

Returns:

maxArguments

public int maxArguments()

Description copied from interface: Aggregate

This method returns maximum number of shapes being passed per Aggregate

Returns:

maxShapes

public int maxShapes()

Description copied from interface: Aggregate

This method returns maximum number of shapes being passed per Aggregate

Returns:

maxIntArrays

public int maxIntArrays()

Description copied from interface: Aggregate

This method returns maximum number of IntArrays being passed per Aggregate

Returns:

maxIntArraySize

public int maxIntArraySize()

Description copied from interface: Aggregate

This method returns maximum length for IntArrays, if any

Returns:

maxIndexArguments

public int maxIndexArguments()

Description copied from interface: Aggregate

This method returns maximum number of IndexArguments per Aggregate

Returns:

maxRealArguments

public int maxRealArguments()

Description copied from interface: Aggregate

This method returns maximum number of real (float/double) per Aggregate

Returns: