@Operator(group="train") public final class NegTrain extends PrimitiveOp
operation
Modifier and Type | Method and Description |
---|---|
static NegTrain |
create(Scope scope,
Operand<Float> wIn,
Operand<Float> wOut,
Operand<Integer> examples,
Operand<Integer> labels,
Operand<Float> lr,
List<Long> vocabCount,
Long numNegativeSamples)
Factory method to create a class wrapping a new NegTrain operation.
|
equals, hashCode, op, toString
public static NegTrain create(Scope scope, Operand<Float> wIn, Operand<Float> wOut, Operand<Integer> examples, Operand<Integer> labels, Operand<Float> lr, List<Long> vocabCount, Long numNegativeSamples)
scope
- current scopewIn
- input word embedding.wOut
- output word embedding.examples
- A vector of word ids.labels
- A vector of word ids.lr
- vocabCount
- Count of words in the vocabulary.numNegativeSamples
- Number of negative samples per example.Copyright © 2022. All rights reserved.