Class

ai.minxiao.ds4s.core.dl4j.embed

W2V

Related Doc: package embed

Permalink

class W2V extends Serializable

W2V

Word2Vec: an Implementation of Skip-Gram Model with Negative Sampling,

Annotations
@SerialVersionUID()
Note

complexity: O = E * T * Q where Q = C * (D + D * log2(V)), E=epochs, T=total-words-of-training-data, C=max-window-size, D=embedding-size, V=vocabulary-size

,

reference: Tomas Mikolov, Kai Chen, Greg Corrado and Jeffrey Dean. Efficient Estimation of Word Representations in Vector Space. In NIPS, 2013.

Linear Supertypes
Serializable, Serializable, AnyRef, Any
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. W2V
  2. Serializable
  3. Serializable
  4. AnyRef
  5. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Instance Constructors

  1. new W2V(minWordFrequency: Int = 5, layerSize: Int = 100, windowSize: Int = 5, epochs: Int = 1, batchSize: Int = 16, iterations: Int = 1, sampling: Double = 0, learningRate: Double = 1E-1, minLearningRate: Double = 1E-6, useAdaGrad: Boolean = false, tokenizerFactory: TokenizerFactory = new DefaultTokenizerFactory, tokenPreProcessor: TokenPreProcess = new CommonPreprocessor, lookupTable: Option[InMemoryLookupTable[VocabWord]] = None, vocabCache: Option[AbstractCache[VocabWord]] = None, seed: Long = 2018L)

    Permalink

    minWordFrequency

    min word frequency, default=5

    layerSize

    word vector dimensionality, default=100

    windowSize

    window size, default=5

    epochs

    training epochs, default=1

    batchSize

    batch size, default=12

    sampling

    sampling rate, default=0(no sampling)

    learningRate

    learning rate, default=1E-1

    minLearningRate

    mini learning rate, default=1E-6

    useAdaGrad

    whether to use AdaGrad, default=false

    tokenizerFactory

    tokenizer factory, default=new DefaultTokenizerFactory

    tokenPreProcessor

    token preprocessor, default=new CommonPreprocessor

    seed

    seed for random generator, default=2018

Value Members

  1. final def !=(arg0: Any): Boolean

    Permalink
    Definition Classes
    AnyRef → Any
  2. final def ##(): Int

    Permalink
    Definition Classes
    AnyRef → Any
  3. final def ==(arg0: Any): Boolean

    Permalink
    Definition Classes
    AnyRef → Any
  4. final def asInstanceOf[T0]: T0

    Permalink
    Definition Classes
    Any
  5. def build(): Word2Vec

    Permalink
  6. def clone(): AnyRef

    Permalink
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  7. final def eq(arg0: AnyRef): Boolean

    Permalink
    Definition Classes
    AnyRef
  8. def equals(arg0: Any): Boolean

    Permalink
    Definition Classes
    AnyRef → Any
  9. def finalize(): Unit

    Permalink
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  10. final def getClass(): Class[_]

    Permalink
    Definition Classes
    AnyRef → Any
  11. def hashCode(): Int

    Permalink
    Definition Classes
    AnyRef → Any
  12. final def isInstanceOf[T0]: Boolean

    Permalink
    Definition Classes
    Any
  13. final val iterations: Int

    Permalink
  14. final def ne(arg0: AnyRef): Boolean

    Permalink
    Definition Classes
    AnyRef
  15. final def notify(): Unit

    Permalink
    Definition Classes
    AnyRef
  16. final def notifyAll(): Unit

    Permalink
    Definition Classes
    AnyRef
  17. final def synchronized[T0](arg0: ⇒ T0): T0

    Permalink
    Definition Classes
    AnyRef
  18. def toString(): String

    Permalink
    Definition Classes
    AnyRef → Any
  19. final def wait(): Unit

    Permalink
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  20. final def wait(arg0: Long, arg1: Int): Unit

    Permalink
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  21. final def wait(arg0: Long): Unit

    Permalink
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

Inherited from Serializable

Inherited from Serializable

Inherited from AnyRef

Inherited from Any

Ungrouped