public class AggregateSkipGram extends BaseAggregate
arguments, finalResult, indexingArguments, intArrayArguments, realArguments, shapes
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
|
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() |
getArguments, getFinalResult, getIndexingArguments, getIntArrayArguments, getRealArguments, getRequiredBatchMemorySize, getShapes, setFinalResult
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)
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)
public AggregateSkipGram(int w1, int w2, int[] codes, int[] points, int negSamples, double lr, int vectorLength)
w1
- w2
- lr
- vectorLength
- public int getSharedMemorySize()
public int getThreadsPerInstance()
public String name()
public int opNum()
public int maxArguments()
Aggregate
public int maxShapes()
Aggregate
public int maxIntArrays()
Aggregate
public int maxIntArraySize()
Aggregate
public int maxIndexArguments()
Aggregate
public int maxRealArguments()
Aggregate
Copyright © 2017. All rights reserved.