org.deeplearning4j.spark.models.embeddings.glove.cooccurrences

Class CoOccurrenceCalculator

java.lang.Object

org.deeplearning4j.spark.models.embeddings.glove.cooccurrences.CoOccurrenceCalculator

All Implemented Interfaces:

java.io.Serializable, org.apache.spark.api.java.function.Function<Pair<java.util.List<java.lang.String>,java.util.concurrent.atomic.AtomicLong>,CounterMap<java.lang.String,java.lang.String>>

public class CoOccurrenceCalculator
extends java.lang.Object
implements org.apache.spark.api.java.function.Function<Pair<java.util.List<java.lang.String>,java.util.concurrent.atomic.AtomicLong>,CounterMap<java.lang.String,java.lang.String>>

Calculate co occurrences based on tokens

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
CoOccurrenceCalculator(boolean symmetric, org.apache.spark.broadcast.Broadcast<VocabCache<VocabWord>> vocab, int windowSize)

Method Summary

Modifier and Type	Method and Description
CounterMap<java.lang.String,java.lang.String>	call(Pair<java.util.List<java.lang.String>,java.util.concurrent.atomic.AtomicLong> pair)

Methods inherited from class java.lang.Object

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

Constructor Detail

CoOccurrenceCalculator

public CoOccurrenceCalculator(boolean symmetric,
                              org.apache.spark.broadcast.Broadcast<VocabCache<VocabWord>> vocab,
                              int windowSize)

Method Detail

call

public CounterMap<java.lang.String,java.lang.String> call(Pair<java.util.List<java.lang.String>,java.util.concurrent.atomic.AtomicLong> pair)
                                                   throws java.lang.Exception

Specified by:

call in interface org.apache.spark.api.java.function.Function<Pair<java.util.List<java.lang.String>,java.util.concurrent.atomic.AtomicLong>,CounterMap<java.lang.String,java.lang.String>>

Throws:

java.lang.Exception