Interface EmbeddingInitializer

    • 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