【发布时间】:2021-12-04 17:40:16
【问题描述】:
这是我第一次写文章。我通常很快就能找到我想要的东西,但这次我没有运气。
我的问题很简单,我有一个包含 True 和 False 值的目标列。基本上,这是一个二元分类问题。我想知道如何使用 Precision(而不是指标:Accuracy)优化我的 CNN?
顺便说一句,这是行不通的:
model.compile(loss='binary_crossentropy', optimizer=optm, metrics=['precision'])
这是我的代码:
model = Sequential()
model.add(Dense(64,name = 'Primera', input_dim=8, activation='relu'))
model.add(Dense(32 ,name = 'Segunda'))
model.add(Dense(1,name = 'Tercera', activation='sigmoid'))
from tensorflow.keras import optimizers
optm = optimizers.Adam(learning_rate=0.001, beta_1=0.9, beta_2=0.999, amsgrad=False)
model.compile(loss='binary_crossentropy', optimizer=optm, metrics=['accuracy'])
model.summary()
history = model.fit(trainX, trainY,
epochs=1000,
batch_size=16,
validation_split=0.1,
verbose=1)
谢谢!
【问题讨论】:
标签: python tensorflow keras binary classification