Keras API 新旧对照表

最新推荐文章于 2024-03-28 07:06:21 发布

原创最新推荐文章于 2024-03-28 07:06:21 发布 · 1.9k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#keras #深度学习 #人工智能

研发管理同时被 2 个专栏收录

7 篇文章

订阅专栏

算法

3 篇文章

订阅专栏

本文介绍如何将Keras LSTM层的调用从旧版API迁移到Keras2 API，包括参数更新和转换对照表，帮助理解并顺利进行代码迁移。

UserWarning: Update your `LSTM` call to the Keras 2 API: `LSTM(128, name="LSTMSpectral", kernel_regularizer=<keras.reg..., recurrent_regularizer=<keras.reg..., implementation=2)`

LSTM(128,name='LSTMSpectral',consume_less='gpu',W_regularizer=l2(0.0001),U_regularizer=l2(0.0001))(LSTMInput)

开始有些不明所以然，查了下面的表，OK。


legacy_recurrent_support = generate_legacy_interface(
    allowed_positional_args=['units'],
    conversions=[('output_dim', 'units'),
                 ('init', 'kernel_initializer'),
                 ('inner_init', 'recurrent_initializer'),
                 ('inner_activation', 'recurrent_activation'),
                 ('W_regularizer', 'kernel_regularizer'),
                 ('b_regularizer', 'bias_regularizer'),
                 ('U_regularizer', 'recurrent_regularizer'),
                 ('dropout_W', 'dropout'),
                 ('dropout_U', 'recurrent_dropout'),
                 ('consume_less', 'implementation')],
    value_conversions={'consume_less': {'cpu': 0,
                                        'mem': 1,
                                        'gpu': 2}},
    preprocessor=recurrent_args_preprocessor)