微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

Keras 分类模型布局

如何解决Keras 分类模型布局

我想使用 Keras(或任何其他 NN 库,如果有更好的)来对数据进行分类

输入看起来像这样:

train_x = np.array([ [[1,3,5,7],[4,6,8,10]],[[6,1,9,[[1,[5,'''.
                       .
                       .'''
                  ])

输出如下:

train_y = np.array([1,1'''...'''])

这是我目前的其余代码

model = Sequential()

model.add(Flatten(input_shape=(2,4))) 
model.add(Dense(20,activation='relu'))
model.add(Dense(1,activation='softmax'))
model.summary()

#compile model using accuracy to measure model performance
model.compile(optimizer='adam',loss='categorical_crossentropy',metrics=['accuracy'])

#train model
early_stopping_monitor = EarlyStopping(patience=3)
model.fit(train_x,train_y,epochs=30,validation_split=0.2,callbacks [early_stopping_monitor])

最初,第一层也是一个 Dense 层,但是这给了我一个错误消息,因为输出是二维的,所以我用一个 Flatten 层替换了它。

控制台输出如下所示:

Console output

显然,它不起作用,我怀疑这是因为模型及其层的布局。有人知道如何为这种情况正确构建模型吗?

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。