【发布时间】:2020-05-12 23:14:39
【问题描述】:
我正在训练一个包含两个类别的 CNN 模型进行预测。我知道它给了我一个类和另一个类的概率,我也知道我可以得到预测的标签,但我没有给出结果。每个评估输入的输出总和不应该等于 1.0 吗?例如:
[[0.2858745 0.85059494]
[0.2858745 0.85059494]
[0.6040499 0.5927084 ]
[0.8403308 0.291448 ]
[0.04195209 0.95504093]
[0.79433376 0.21279709]
[0.79433376 0.21279709]
[0.01326967 0.9891382 ]
[0.0153821 0.9867737 ]
[0.79433376 0.21279709]
[0.01617167 0.98520505]
[0.01351487 0.98596036]
[0.01473185 0.9846144 ]
[0.00896762 0.9899838 ]
[0.00936404 0.9893628 ]]
有什么我没有得到的吗?
我的代码:
model_05_01 = Sequential()
model_05_01.add(Conv1D(filters=16, kernel_size=12,
input_shape=(x_train.shape[1], 1)))
model_05_01.add(MaxPooling1D(pool_size=4))
model_05_01.add(Conv1D(filters=32, kernel_size=12))
model_05_01.add(MaxPooling1D(pool_size=4))
model_05_01.add(Conv1D(filters=16, kernel_size=12))
model_05_01.add(MaxPooling1D(pool_size=4))
model_05_01.add(Flatten())
model_05_01.add(Dense(16, activation='relu'))
model_05_01.add(Dense(2, activation='sigmoid'))
model_05_01.compile(loss='logcosh', optimizer='adam',
metrics=['accuracy'])
【问题讨论】:
标签: machine-learning keras deep-learning conv-neural-network