org.deeplearning4j.nn.weights.embeddings

Interface EmbeddingInitializer

All Superinterfaces:

Serializable

All Known Implementing Classes:

ArrayEmbeddingInitializer

public interface EmbeddingInitializer
extends Serializable

An interface implemented by things like Word2Vec etc that allows them to be used as weight

Method Summary

Modifier and Type	Method and Description
boolean	jsonSerializable()
void	loadWeightsInto(INDArray array) Load the weights into the specified INDArray
int	vectorSize()
long	vocabSize()

Method Detail

loadWeightsInto

void loadWeightsInto(INDArray array)

Load the weights into the specified INDArray

Parameters:

array - Array of shape [vocabSize, vectorSize]

vocabSize

long vocabSize()

Returns:

Size of the vocabulary

vectorSize

int vectorSize()

Returns:

Size of each vector

jsonSerializable

boolean jsonSerializable()

Returns:

True if the embedding initializer can be safely serialized as JSON