Class StaticWord2Vec.Builder

    • Constructor Detail

      • Builder

        public Builder​(org.nd4j.linalg.compression.AbstractStorage<Integer> storage,
                       VocabCache<VocabWord> vocabCache)
        Parameters:
        storage - AbstractStorage implementation, key has to be Integer, index of vocabWords
        vocabCache - VocabCache implementation, which will be used to lookup word indexes
    • Method Detail

      • setCachePerDevice

        public StaticWord2Vec.Builder setCachePerDevice​(long bytes)
        This method lets you to define if decompressed values will be cached, to avoid excessive decompressions. If bytes == 0 - no cache will be used.
        Parameters:
        bytes -
        Returns:
      • build

        public StaticWord2Vec build()
        This method returns Static Word2Vec implementation, which is suitable for tasks like neural nets feeding.
        Returns: