org.tensorflow.op.core

Class NegTrain

java.lang.Object

org.tensorflow.op.PrimitiveOp

org.tensorflow.op.core.NegTrain

All Implemented Interfaces:

Op

@Operator
public final class NegTrain
extends PrimitiveOp

Training via negative sampling.

Field Summary

Fields inherited from class org.tensorflow.op.PrimitiveOp

operation

Method Summary

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 to wrap a new NegTrain operation to the graph.

Methods inherited from class org.tensorflow.op.PrimitiveOp

equals, hashCode, toString

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Method Detail

create

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)

Factory method to create a class to wrap a new NegTrain operation to the graph.

Parameters:

scope - current graph scope

wIn - 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.

Returns:

a new instance of NegTrain