【发布时间】:2018-11-05 10:42:42
【问题描述】:
我有一个包含 65 个特征和 2 个标签的数据集,并且想要执行二进制分类。
问题是测试损失不会提高,测试精度也不会提高。
训练损失与测试损失
代码
model = Sequential()
model = Sequential()
model.add(Dense(10, activation='relu', input_dim=n_features))
model.add(Dropout(0.5))
model.add(Dense(2, activation='softmax'))
model.compile(loss='binary_crossentropy', optimizer='adam',metrics=['acc'])
数据集高度不平衡 (1:30)
如何改进我的模型?
【问题讨论】:
标签: python machine-learning keras