Have a question? Connect with the community at the TensorFlow Forum Visit Forum

NegTrain

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