美文网首页
多输入和共享层的概念

多输入和共享层的概念

作者: 光光小丸子 | 来源:发表于2018-11-19 14:20 被阅读0次

    这里主要实现以下多输入和共享层的概念

    '''
    from keras.layers import Input, LSTM, Dense, merge,Concatenate
    from keras.models import Model

    tweet_a = Input(shape=(140, 256))
    tweet_b = Input(shape=(140, 256))

    shared_lstm = LSTM(64)

    encoded_a = shared_lstm(tweet_a)
    encoded_b = shared_lstm(tweet_b)

    print('int0',shared_lstm.get_input_at(0))
    print('int1',shared_lstm.get_input_at(1))

    print('out0',shared_lstm.get_output_at(0))

    print('out1',shared_lstm.get_output_at(1))
    merged_vector = Concatenate(axis=-1)([encoded_a,encoded_b])

    predictions = Dense(1, activation='sigmoid')(merged_vector)

    model = Model(inputs=[tweet_a, tweet_b], outputs=predictions)

    model.compile(optimizer='rmsprop',
    loss='binary_crossentropy',
    metrics=['accuracy'])

    print(model.summary())
    '''

    相关文章

      网友评论

          本文标题:多输入和共享层的概念

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