org.deeplearning4j.spark.models.embeddings.glove

Class Glove

java.lang.Object

org.deeplearning4j.spark.models.embeddings.glove.Glove

All Implemented Interfaces:

java.io.Serializable

public class Glove
extends java.lang.Object
implements java.io.Serializable

Spark glove

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
Glove(boolean symmetric, int windowSize, int iterations)
Glove(java.lang.String tokenizerFactoryClazz, boolean symmetric, int windowSize, int iterations)

Method Summary

Modifier and Type	Method and Description
Pair<VocabCache<VocabWord>,GloveWeightLookupTable>	train(org.apache.spark.api.java.JavaRDD<java.lang.String> rdd) Train on the corpus

Methods inherited from class java.lang.Object

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

Constructor Detail

Glove

public Glove(java.lang.String tokenizerFactoryClazz,
             boolean symmetric,
             int windowSize,
             int iterations)

Parameters:

tokenizerFactoryClazz - the fully qualified class name of the tokenizer

symmetric - whether the co occurrence counts should be symmetric

windowSize - the window size for co occurrence

iterations - the number of iterations

Glove

public Glove(boolean symmetric,
             int windowSize,
             int iterations)

Parameters:

symmetric - whether the co occurrence counts should be symmetric

windowSize - the window size for co occurrence

iterations - the number of iterations

Method Detail

train

public Pair<VocabCache<VocabWord>,GloveWeightLookupTable> train(org.apache.spark.api.java.JavaRDD<java.lang.String> rdd)
                                                         throws java.lang.Exception

Train on the corpus

Parameters:

rdd - the rdd to train

Returns:

the vocab and weights

Throws:

java.lang.Exception