【发布时间】:2020-10-28 08:52:40
【问题描述】:
我开始使用 Keras 和那些模型,但我认为我还没有掌握所有这些。我摆弄了很多东西,但只会让情况变得更糟。我的目标是达到高于 91% 的准确度(91% 的准确度是该模型的输出)。 这是代码:
import numpy as np
import tensorflow as tf
input_size = 10
output_size = 2
hidden_layer_size = 500
model = tf.keras.Sequential([
tf.keras.layers.Dense(hidden_layer_size, activation='relu'),
tf.keras.layers.Dense(hidden_layer_size, activation='relu'),
tf.keras.layers.Dense(output_size, activation='softmax') # output layer
])
model.compile(optimizer='adam', loss='sparse_categorical_crossentropy', metrics=['accuracy'])
batch_size = 50
max_epochs = 100
early_stopping = tf.keras.callbacks.EarlyStopping(patience=5)
model.fit(train_inputs,
train_targets,
batch_size=batch_size,
epochs=max_epochs,
callbacks=[early_stopping],
validation_data=(validation_inputs, validation_targets),
verbose = 2
)
我不知道我是否在这里格式化了这个,我刚刚创建了这个帐户,我是这方面的初学者。
【问题讨论】:
标签: python python-3.x tensorflow keras tensorflow2.0