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 boolean
jsonSerializable()
void
loadWeightsInto(INDArray array)
Load the weights into the specified INDArrayint
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
-
-