org.deeplearning4j.spark.models.embeddings.word2vec

Class FirstIterationFunction

java.lang.Object

org.datavec.spark.transform.BaseFlatMapFunctionAdaptee<java.util.Iterator<scala.Tuple2<java.util.List<VocabWord>,java.lang.Long>>,java.util.Map.Entry<VocabWord,org.nd4j.linalg.api.ndarray.INDArray>>

org.deeplearning4j.spark.models.embeddings.word2vec.FirstIterationFunction

All Implemented Interfaces:

java.io.Serializable, org.apache.spark.api.java.function.FlatMapFunction<java.util.Iterator<scala.Tuple2<java.util.List<VocabWord>,java.lang.Long>>,java.util.Map.Entry<VocabWord,org.nd4j.linalg.api.ndarray.INDArray>>

public class FirstIterationFunction
extends org.datavec.spark.transform.BaseFlatMapFunctionAdaptee<java.util.Iterator<scala.Tuple2<java.util.List<VocabWord>,java.lang.Long>>,java.util.Map.Entry<VocabWord,org.nd4j.linalg.api.ndarray.INDArray>>

See Also:

Serialized Form

Field Summary

Fields inherited from class org.datavec.spark.transform.BaseFlatMapFunctionAdaptee

adapter

Constructor Summary

Constructors

Constructor and Description
FirstIterationFunction(org.apache.spark.broadcast.Broadcast<java.util.Map<java.lang.String,java.lang.Object>> word2vecVarMapBroadcast, org.apache.spark.broadcast.Broadcast<double[]> expTableBroadcast, org.apache.spark.broadcast.Broadcast<VocabCache<VocabWord>> vocabCacheBroadcast)

Method Summary

Methods inherited from class org.datavec.spark.transform.BaseFlatMapFunctionAdaptee

call

Methods inherited from class java.lang.Object

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

Constructor Detail

FirstIterationFunction

public FirstIterationFunction(org.apache.spark.broadcast.Broadcast<java.util.Map<java.lang.String,java.lang.Object>> word2vecVarMapBroadcast,
                              org.apache.spark.broadcast.Broadcast<double[]> expTableBroadcast,
                              org.apache.spark.broadcast.Broadcast<VocabCache<VocabWord>> vocabCacheBroadcast)