首页 > 解决方案 > 我为 LSTM 和 GRU 尝试了这段代码。但是 GRU 不工作

问题描述

我不确定这段代码哪里错了。我的目标是使用 LSTM 和 GRU 训练我的数据集进行二进制分类。
[输出带有模块包装器和 GRU 未执行请检查图像][1]

    
        #BUILD THE MODEL
        top_words = 10000
        embedding_vecor_length = 32
        model = Sequential()
        model.add(Embedding(top_words, embedding_vecor_length, input_length=X.shape[1]))
        #model.add(Dropout(0.2))
        model.add(GRU(100,dropout=0.2, recurrent_dropout=0.2, return_sequences=True))
        model.add(LSTM(100,dropout=0.2, recurrent_dropout=0.2))
        #model.add(Dropout(0.2))
        model.add(Dense(1, activation='sigmoid'))
        model.add(Dense(1, activation='sigmoid'))
        model.compile(loss='binary_crossentropy', optimizer='Adam',
                      metrics=['accuracy'])
        print(model.summary())   
        model.summary()


    ```


  [1]: https://i.stack.imgur.com/14pyl.jpg

标签: pythonneural-networklstm

解决方案


推荐阅读