keras的蚊香实例

关键代码

model = Sequential()
model.add(Dense(units=8,activation='relu',input_dim=2))
model.add(Dense(units=8,activation='relu'))
model.add(Dense(units=8,activation='relu'))
model.add(Dense(units=1,activation='sigmoid'))


model.compile(loss='mean_squared_error',optimizer=SGD(lr=0.05),metrics=['accuracy'])

代码

原图

预测图象

预测率