Interface EmbeddingInitializer
-
- All Superinterfaces:
Serializable
- All Known Implementing Classes:
ArrayEmbeddingInitializer
public interface EmbeddingInitializer extends Serializable
-
-
Method Summary
All Methods Instance Methods Abstract Methods Modifier and Type Method Description booleanjsonSerializable()voidloadWeightsInto(INDArray array)Load the weights into the specified INDArrayintvectorSize()longvocabSize()
-
-
-
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
-
-