org.deeplearning4j.spark.models.embeddings.common

Class BaseWord2VecFunction

java.lang.Object

org.deeplearning4j.spark.models.embeddings.common.BaseWord2VecFunction

All Implemented Interfaces:

Serializable

Direct Known Subclasses:

DotProduct, GetVector, SaxpyMap

public class BaseWord2VecFunction
extends Object
implements Serializable

Base word2vec function for hazelcast. Stores how the lookup table should be accessed

Author:

Adam Gibson

See Also:

Serialized Form

Field Summary

Fields

Modifier and Type	Field and Description
protected String	from
protected int	fromIndex
static String	SYN0
static String	SYN1
static String	SYN1_NEGATIVE
protected String	to
protected int	toIndex

Constructor Summary

Constructors

Constructor and Description
BaseWord2VecFunction(int fromIndex, int toIndex, String from, String to)

Method Summary

Modifier and Type	Method and Description
protected org.nd4j.linalg.api.ndarray.INDArray	getFrom(InMemoryLookupTable inMemoryLookupTable)
protected org.nd4j.linalg.api.ndarray.INDArray	getTo(InMemoryLookupTable inMemoryLookupTable)

Methods inherited from class java.lang.Object

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

Field Detail

fromIndex

protected int fromIndex

toIndex

protected int toIndex

from

protected String from

to

protected String to

SYN1

public static final String SYN1

See Also:

Constant Field Values

SYN0

public static final String SYN0

See Also:

Constant Field Values

SYN1_NEGATIVE

public static final String SYN1_NEGATIVE

See Also:

Constant Field Values

Constructor Detail

BaseWord2VecFunction

public BaseWord2VecFunction(int fromIndex,
                            int toIndex,
                            String from,
                            String to)

Method Detail

getFrom

protected org.nd4j.linalg.api.ndarray.INDArray getFrom(InMemoryLookupTable inMemoryLookupTable)

getTo

protected org.nd4j.linalg.api.ndarray.INDArray getTo(InMemoryLookupTable inMemoryLookupTable)