美文网首页keras深度学习模型
5 使用1D卷积的序列分类

5 使用1D卷积的序列分类

作者: readilen | 来源:发表于2017-05-29 10:47 被阅读1437次
    from keras.models import Sequential
    from keras.layers import Dense, Dropout
    from keras.layers import Embedding
    from keras.layers import Conv1D, GlobalAveragePooling1D, MaxPooling1D
    
    model = Sequential()
    model.add(Conv1D(64, 3, activation='relu', input_shape=(seq_length, 100)))
    model.add(Conv1D(64, 3, activation='relu'))
    model.add(MaxPooling1D(3))
    model.add(Conv1D(128, 3, activation='relu'))
    model.add(Conv1D(128, 3, activation='relu'))
    model.add(GlobalAveragePooling1D())
    model.add(Dropout(0.5))
    model.add(Dense(1, activation='sigmoid'))
    
    model.compile(loss='binary_crossentropy',
                  optimizer='rmsprop',
                  metrics=['accuracy'])
    
    model.fit(x_train, y_train, batch_size=16, epochs=10)
    score = model.evaluate(x_test, y_test, batch_size=16)
    

    相关文章

      网友评论

      • 928dd75803e9:您好,我想问下您用的x_train, y_train的shape都是多大,运行的时候遇到了问题,感谢~
        readilen:@月亮不开花 16
      • 4f381a7bb087:这就是官方教程里的例子啊

      本文标题:5 使用1D卷积的序列分类

      本文链接:https://www.haomeiwen.com/subject/gjukfxtx.html