org.deeplearning4j.spark.models.embeddings.word2vec

Class MapToPairFunction

java.lang.Object

org.deeplearning4j.spark.models.embeddings.word2vec.MapToPairFunction

All Implemented Interfaces:

Serializable, org.apache.spark.api.java.function.Function<Map.Entry<VocabWord,org.nd4j.linalg.api.ndarray.INDArray>,Pair<VocabWord,org.nd4j.linalg.api.ndarray.INDArray>>

public class MapToPairFunction
extends Object
implements org.apache.spark.api.java.function.Function<Map.Entry<VocabWord,org.nd4j.linalg.api.ndarray.INDArray>,Pair<VocabWord,org.nd4j.linalg.api.ndarray.INDArray>>

Author:

jeffreytang

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
MapToPairFunction()

Method Summary

Modifier and Type	Method and Description
Pair<VocabWord,org.nd4j.linalg.api.ndarray.INDArray>	call(Map.Entry<VocabWord,org.nd4j.linalg.api.ndarray.INDArray> pair)

Methods inherited from class java.lang.Object

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

Constructor Detail

MapToPairFunction

public MapToPairFunction()

Method Detail

call

public Pair<VocabWord,org.nd4j.linalg.api.ndarray.INDArray> call(Map.Entry<VocabWord,org.nd4j.linalg.api.ndarray.INDArray> pair)

Specified by:

call in interface org.apache.spark.api.java.function.Function<Map.Entry<VocabWord,org.nd4j.linalg.api.ndarray.INDArray>,Pair<VocabWord,org.nd4j.linalg.api.ndarray.INDArray>>