public class AggregateCBOW extends BaseAggregate
arguments, finalResult, indexingArguments, intArrayArguments, realArguments, shapes
Constructor and Description |
---|
AggregateCBOW(INDArray syn0,
INDArray syn1,
INDArray syn1Neg,
INDArray expTable,
INDArray negTable,
int wordIdx,
int[] idxSyn0,
int[] idxSyn1,
int[] codes,
int negativeRounds,
int ngStarter,
int vectorLength,
double alpha,
long nextRandom,
int vocabSize)
Default constructor for CBOW implementation wrapper
|
AggregateCBOW(INDArray syn0,
INDArray syn1,
INDArray syn1Neg,
INDArray expTable,
INDArray negTable,
int wordIdx,
int[] idxSyn0,
int[] idxSyn1,
int[] codes,
int negativeRounds,
int ngStarter,
int vectorLength,
double alpha,
long nextRandom,
int vocabSize,
int numLabels,
boolean trainWords,
INDArray inferenceVector)
Optional constructor for ParagraphVectors PV-DM implementation
|
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 AggregateCBOW(@NonNull INDArray syn0, INDArray syn1, INDArray syn1Neg, @NonNull INDArray expTable, INDArray negTable, int wordIdx, int[] idxSyn0, int[] idxSyn1, int[] codes, int negativeRounds, int ngStarter, int vectorLength, double alpha, long nextRandom, int vocabSize, int numLabels, boolean trainWords, INDArray inferenceVector)
syn0
- syn1
- syn1Neg
- expTable
- negTable
- wordIdx
- idxSyn0
- idxSyn1
- codes
- negativeRounds
- ngStarter
- vectorLength
- alpha
- nextRandom
- vocabSize
- numLabels
- trainWords
- public AggregateCBOW(@NonNull INDArray syn0, INDArray syn1, INDArray syn1Neg, @NonNull INDArray expTable, INDArray negTable, int wordIdx, int[] idxSyn0, int[] idxSyn1, int[] codes, int negativeRounds, int ngStarter, int vectorLength, double alpha, long nextRandom, int vocabSize)
syn0
- syn1
- syn1Neg
- expTable
- negTable
- wordIdx
- idxSyn0
- idxSyn1
- codes
- negativeRounds
- ngStarter
- vectorLength
- alpha
- nextRandom
- vocabSize
- 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
public int getSharedMemorySize()
Aggregate
public int getThreadsPerInstance()
Aggregate
Copyright © 2016. All Rights Reserved.