首页 > 解决方案 > ValueError:未知激活函数:LeakyReLU

问题描述

我已经构建、安装并保存了以下模型:

def build_Lstm_model(train_x,train_y,test_x,test_y):
        inp = Input(shape=(train_x.shape[1],train_x.shape[2]))
        rnn_1st_model = LSTM(units=60, return_sequences=True,recurrent_dropout=0.1)(inp)
        rnn_2nd_model = LSTM(units=60,recurrent_dropout=0.1)(rnn_1st_model)
        dense_layer = Dense(128)(rnn_2nd_model)
        drop_out = Dropout(0.2)(dense_layer)
        output = Dense(1, activation= LeakyReLU(alpha=0.1),name="class")(drop_out)
        model = Model(inp, output)
        callbacks = [ReduceLROnPlateau(monitor='val_loss', patience=4, verbose=1, factor=0.6),
                     EarlyStopping(monitor='val_loss', patience=20),
                     ModelCheckpoint(filepath='model_LSTM.h5', monitor='val_loss', save_best_only=True)]
        model.summary()
        model.compile(loss='binary_crossentropy', optimizer="adam", metrics=['accuracy'])
        history = model.fit(train_x, train_y, 
              epochs = 8, 
              batch_size = 128, 
              validation_data=(test_x,  test_y), 
              callbacks=callbacks)
        return history,model

我正在使用 load_model 函数进行部署,现在出现以下错误:

ValueError: Unknown activation function: LeakyReLU

标签: pythonkerasdeep-learninglstm

解决方案


推荐阅读