DL4J中文文档/Keras模型导入/包装层

KerasBidirectional

[源码]

从Keras双向层包装器构建DL4J双向层

KerasBidirectional

public KerasBidirectional(Integer kerasVersion) throws UnsupportedKerasConfigurationException 

image.gif

从KerasLayer传递构造函数

  • 参数 kerasVersion Keras主版本
  • 抛出 UnsupportedKerasConfigurationException 不支持的 Keras 配置

getUnderlyingRecurrentLayer

public Layer getUnderlyingRecurrentLayer() 

image.gif

获取底层循环层

public KerasBidirectional(Map<String, Object> layerConfig)
            throws InvalidKerasConfigurationException, UnsupportedKerasConfigurationException
image.gif

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

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

getBidirectionalLayer

public Bidirectional getBidirectionalLayer() 

image.gif

获取 DL4J 的双向层。

  • 返回双向层

getOutputType

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

image.gif

获取层输出类型

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

getNumParams

public int getNumParams() 

image.gif

返回层中可训练的参数

getInputPreprocessor

public InputPreProcessor getInputPreprocessor(InputType... inputType) throws InvalidKerasConfigurationException 

image.gif

为给定的输入类型获取适合的DL4J输入预处理器

  • 参数 inputType 输入类型数组
  • 返回 DL4J 输入预处理器
  • 抛出 InvalidKerasConfigurationException 无效的 Keras 配置
  • 查看 org.deeplearning4j.nn.conf.InputPreProcessor

setWeights

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

image.gif

设置层的权重

  • 参数 weights 权重映射。
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容