【发布时间】:2021-06-17 07:32:41
【问题描述】:
'MLP'
'在训练数据集上定义和拟合模型'
def fit_model(trainX, trainy, testX, testy):
'define model'
model = Sequential()
model.add(Dense(5, input_dim=2, activation='relu', kernel_initializer='he_uniform'))
model.add(Dense(5, activation='relu', kernel_initializer='he_uniform'))
model.add(Dense(3, activation='softmax'))
'compile model'
model.compile(loss='categorical_crossentropy', optimizer='sgd', metrics=['acc'])
'fit model'
history = model.fit(trainX, trainy, validation_data=(testX, testy), epochs=100, batch_size=66, verbose=0)
return model, history
# fit model on train dataset
model, history = fit_model(trainX, trainy, testX, testy)
# evaluate model behavior
summarize_model(model, history, trainX, trainy, testX, testy)
# save model to file
model.save('model.h5')
【问题讨论】:
-
请发布格式正确的minimal reproducible example - 请参阅How to Ask。
标签: python keras tensor valueerror mlp