模型预训练 自编码器AE 去噪自编码器 denoiseAE 原码keras Tensorflow 初始化模型参数 模型参数微调

Keras Tensorflow构建自编码器AE原码

使用自编码器AE预训练模型，构建自编码器AE初始化网络参数，然后再微调模型参数。

from keras import models
from keras import layers


def build_pre_normalAE(data_dim, X_train, epoch_pretrain=25, hidDim=[100,140]):
    ae = models.Sequential()
    ae.add(layers.Dense(hidDim[0], activation='tanh', input_shape=(data_dim, )))
    ae.add(layers.Dense(hidDim[1], activation='tanh'))
    ae.add(layers.Dense(hidDim[0], activation='tanh'))