【发布时间】:2017-07-28 16:02:19
【问题描述】:
您将如何在 keras 中为回归问题创建和显示准确度指标,例如在将预测四舍五入到最接近的整数类之后?
虽然准确度本身并没有按惯例有效地定义回归问题,但要确定数据的序数类/标签,将问题视为回归是合适的。但是,计算一个准确度指标会很方便,无论是 kappa 还是其他类似的指标。这是要修改的基本 keras 样板代码。
from keras.models import Sequential
from keras.layers.core import Dense, Activation
model = Sequential()
model.add(Dense(10, 64))
model.add(Activation('tanh'))
model.add(Dense(64, 1))
model.compile(loss='mean_absolute_error', optimizer='rmsprop')
model.fit(X_train, y_train, nb_epoch=20, batch_size=16)
score = model.evaluate(X_test, y_test, batch_size=16)
【问题讨论】:
标签: python machine-learning tensorflow deep-learning keras