首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >Keras词在四元模型中的嵌入

Keras词在四元模型中的嵌入
EN

Stack Overflow用户
提问于 2016-12-28 03:28:04
回答 0查看 1.1K关注 0票数 0

我正在学习coursera神经网络课程,我正在尝试使用python+keras而不是octave来通过作业。

根据前面三个单词,我想预测第四个单词。我的输入文档总共有250个唯一的单词。

该模型应该有一个嵌入层,它将每个单词映射到一个50维向量空间,一个隐藏层,包含200个具有sigmoid激活函数的神经元,以及一个250个单元的输出层,通过softmax激活,将第四个单词的概率与我的词汇表中的单词的概率相等。

我对尺寸有问题。下面是我的代码:

代码语言:javascript
复制
    from keras.models import Sequential

    from keras.layers import Dense, Activation, Embedding


    model = Sequential([Embedding(250,50),
                Dense(200, activation='sigmoid'),
                Dense(250, activation='softmax')

    ])



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

然而,我从来没有编译过这个模型,因为我遇到了以下错误:

代码语言:javascript
复制
    Exception: Input 0 is incompatible with layer dense_1: expected ndim=2, found ndim=3

任何提示都将不胜感激。提前感谢

EN

回答

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/41351057

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档