public static class StaticWord2Vec.Builder
extends java.lang.Object
Constructor and Description |
---|
Builder(org.nd4j.linalg.compression.AbstractStorage<java.lang.Integer> storage,
VocabCache<VocabWord> vocabCache) |
Modifier and Type | Method and Description |
---|---|
StaticWord2Vec |
build()
This method returns Static Word2Vec implementation, which is suitable for tasks like neural nets feeding.
|
StaticWord2Vec.Builder |
setCachePerDevice(long bytes)
This method lets you to define if decompressed values will be cached, to avoid excessive decompressions.
|
public Builder(org.nd4j.linalg.compression.AbstractStorage<java.lang.Integer> storage, VocabCache<VocabWord> vocabCache)
storage
- AbstractStorage implementation, key has to be Integer, index of vocabWordsvocabCache
- VocabCache implementation, which will be used to lookup word indexespublic StaticWord2Vec.Builder setCachePerDevice(long bytes)
bytes
- public StaticWord2Vec build()