NegTrain

Stay organized with collections Save and categorize content based on your preferences.
public final class NegTrain

Training via negative sampling.

Constants

String OP_NAME The name of this op, as known by TensorFlow core engine

Public Methods

static NegTrain
create(Scope scope, Operand<TFloat32> wIn, Operand<TFloat32> wOut, Operand<TInt32> examples, Operand<TInt32> labels, Operand<TFloat32> lr, List<Long> vocabCount, Long numNegativeSamples)
Factory method to create a class wrapping a new NegTrain operation.

Inherited Methods

Constants

public static final String OP_NAME

The name of this op, as known by TensorFlow core engine

Constant Value: "NegTrain"

Public Methods

public static NegTrain create (Scope scope, Operand<TFloat32> wIn, Operand<TFloat32> wOut, Operand<TInt32> examples, Operand<TInt32> labels, Operand<TFloat32> lr, List<Long> vocabCount, Long numNegativeSamples)

Factory method to create a class wrapping a new NegTrain operation.

Parameters
scope current scope
wIn input word embedding.
wOut output word embedding.
examples A vector of word ids.
labels A vector of word ids.
vocabCount Count of words in the vocabulary.
numNegativeSamples Number of negative samples per example.
Returns
  • a new instance of NegTrain