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

 

KerasBidirectional

[源码]

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

KerasBidirectional

public KerasBidirectional(Integer kerasVersion) throws UnsupportedKerasConfigurationException 

从KerasLayer传递构造函数

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

getUnderlyingRecurrentLayer

public Layer getUnderlyingRecurrentLayer() 

获取底层循环层

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

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

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

getBidirectionalLayer

public Bidirectional getBidirectionalLayer() 

获取 DL4J 的双向层。

  • 返回双向层

getOutputType

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

获取层输出类型

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

getNumParams

public int getNumParams() 

返回层中可训练的参数 

getInputPreprocessor

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

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

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

setWeights

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

设置层的权重

  • 参数 weights 权重映射。

猜你喜欢

转载自blog.csdn.net/bewithme/article/details/86536829