开发者社区> 问答> 正文

检查输入时出错:预期dense_1_input有2个维度,但是得到了一个带形状的数组(25000,7

trainData。=(25000, 700, 50),形状如下:

[[[ 0.7095   0.863    0.712   ...  0.02715 -1.305    0.5195 ]
  [-0.66     1.715   -1.934   ...  0.5684   0.754    0.2593 ]
  [-0.3533   2.256   -1.292   ... -0.2708   0.6714  -1.128  ]
  ...
  [ 0.       0.       0.      ...  0.       0.       0.     ]
  [ 0.       0.       0.      ...  0.       0.       0.     ]
  [ 0.       0.       0.      ...  0.       0.       0.     ]]
  ...

trainLabel。=(25000,),,形状如下:

[1. 1. 1. ... 0. 0. 0.]

使用它们来训练MLP模型,我应该做什么?具体代码如下:

def MySimpleMLP(feature=700, vec_size=50):
    auc_roc = LSTM.as_keras_metric(tf.compat.v1.metrics.auc)

    model = Sequential()

    model.add(Dense(32, activation='relu', input_shape=(feature * vec_size,)))
    model.add(Dropout(0.2))
    model.add(Dense(32, activation='relu'))
    model.add(Dropout(0.2))
    model.add(Dense(1, activation='softmax'))

    model.compile(loss="binary_crossentropy", optimizer="adam", metrics=[auc_roc])
    return model

 ......       

 model.fit(trainData, trainLabel, validation_split=0.2, epochs=10, batch_size=64, verbose=2)

我应该如何重塑trainData和trainLabel ?请帮助。 问题来源StackOverflow 地址:/questions/59466678/error-when-checking-input-expected-dense-1-input-to-have-2-dimensions-but-got

展开
收起
kun坤 2019-12-25 09:43:29 716 0
1 条回答
写回答
取消 提交回答
  • def MySimpleMLP(feature=700, vec_size=50): auc_roc = LSTM.as_keras_metric(tf.compat.v1.metrics.auc)

    model = Sequential()
    
    model.add(Dense(32, activation='relu', input_shape=(feature * vec_size,)))
    model.add(Dropout(0.2))
    model.add(Dense(32, activation='relu'))
    model.add(Flatten())
    model.add(Dropout(0.2))
    model.add(Dense(1, activation='softmax'))
    
    model.compile(loss="binary_crossentropy", optimizer="adam", metrics=[auc_roc])
    return model
    

    ......

    model.fit(trainData, trainLabel, validation_split=0.2, epochs=10, batch_size=64, verbose=2)

    2019-12-26 14:32:02
    赞同 展开评论 打赏
问答地址:
问答排行榜
最热
最新

相关电子书

更多
重新定义计算的边界 立即下载
低代码开发师(初级)实战教程 立即下载
阿里巴巴DevOps 最佳实践手册 立即下载