| Package | Description |
|---|---|
| org.deeplearning4j.nn.conf.layers | |
| org.deeplearning4j.nn.weights.embeddings |
| Modifier and Type | Method and Description |
|---|---|
EmbeddingSequenceLayer.Builder |
EmbeddingSequenceLayer.Builder.weightInit(EmbeddingInitializer embeddingInitializer)
Initialize the embedding layer using the specified EmbeddingInitializer - such as a Word2Vec instance
|
EmbeddingLayer.Builder |
EmbeddingLayer.Builder.weightInit(EmbeddingInitializer embeddingInitializer)
Initialize the embedding layer using the specified EmbeddingInitializer - such as a Word2Vec instance
|
| Modifier and Type | Class and Description |
|---|---|
class |
ArrayEmbeddingInitializer
Embedding layer initialization from a specified array
|
| Constructor and Description |
|---|
WeightInitEmbedding(@NonNull EmbeddingInitializer embeddingInitializer) |
WeightInitEmbedding(EmbeddingInitializer serializableInit,
EmbeddingInitializer nonSerializableInit) |
Copyright © 2020. All rights reserved.