【发布时间】:2020-04-26 11:02:08
【问题描述】:
我已经使用 Keras 尝试制作一个基本的神经网络,并且我正在使用一些人口普查数据。该网络应该识别个人是否每年赚“50K”。我的准确度很好,但我也想知道每个实例中有多少是正确的。
所以我要统计的四个参数是:
模型猜测“50K”?
模型猜了多少次“
本应为“50K”多少次?
模型猜测“>50K”的次数是“>50K”?
#define model architecture
model = Sequential()
n_cols = train_x.shape[1]
model.add(Dense(10, activation='relu', input_shape=(n_cols,)))
model.add(Dense(10, activation='relu'))
model.add(Dense(1))
#compile model
model.compile(loss='mean_squared_error',
optimizer='adam',
metrics=['accuracy'])
#fit model on training data
model.fit(train_x, train_y,
nb_epoch=50, verbose=1)
#evaluate model on test data
accuracy = model.evaluate(test_x, test_y, verbose=1)
print("loss: ", accuracy[0]*100)
print("accuracy: ", accuracy[1]*100)
【问题讨论】:
-
请注意,您实际上有一个分类问题,并且您正在将其视为回归问题,这是完全错误的
标签: python tensorflow machine-learning keras