DL4J中文文档/Keras模型导入/嵌入层

KerasEmbedding

[源码]

从Keras导入一个嵌入层

KerasEmbedding

public KerasEmbedding() throws UnsupportedKerasConfigurationException 

image.gif

为单元测试传递构造函数

• 抛出 UnsupportedKerasConfigurationException 不支持的 Keras 配置

getEmbeddingLayer

public EmbeddingSequenceLayer getEmbeddingLayer() 

image.gif

获取嵌入层

public KerasEmbedding(Map<String, Object> layerConfig)
            throws InvalidKerasConfigurationException, UnsupportedKerasConfigurationException 

image.gif

来自解析的keras层配置字典的构造函数。

• 参数 layerConfig 包含keras层配置的字典
• 抛出 InvalidKerasConfigurationException 无效的 Keras 配置
• 抛出 UnsupportedKerasConfigurationException 不支持的无效的 Keras 配置

getOutputType

public InputType getOutputType(InputType... inputType) throws InvalidKerasConfigurationException 

image.gif

获取层输出类型

• 参数 inputType 输入类型数组
• 返回 按输入类型返回输出类型
• 抛出 InvalidKerasConfigurationException 无效的 Keras 配置

getNumParams

public int getNumParams() 

image.gif

返回层中可训练参数的数目。

setWeights

public void setWeights(Map<String, INDArray> weights) throws InvalidKerasConfigurationException 

image.gif

为层设置权重

• 参数 weights 嵌入层权重