org.tensorflow.op.core

Class Skipgram.Options

java.lang.Object

org.tensorflow.op.core.Skipgram.Options

Enclosing class:

Skipgram

public static class Skipgram.Options
extends Object

Optional attributes for Skipgram

Method Summary

Modifier and Type	Method and Description
Skipgram.Options	minCount(Long minCount)
Skipgram.Options	subsample(Float subsample)
Skipgram.Options	windowSize(Long windowSize)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Detail

windowSize

public Skipgram.Options windowSize(Long windowSize)

Parameters:

windowSize - The number of words to predict to the left and right of the target.

minCount

public Skipgram.Options minCount(Long minCount)

Parameters:

minCount - The minimum number of word occurrences for it to be included in the vocabulary.

subsample

public Skipgram.Options subsample(Float subsample)

Parameters:

subsample - Threshold for word occurrence. Words that appear with higher frequency will be randomly down-sampled. Set to 0 to disable.