在 multiclass classification 中寻找 class 的概率
Finding probability of a class in multiclass classification
我制作了一个 Tf/Keras CNN 模型，它是一个二元分类器。该模型对图像属于 Class A 还是 Class B 进行分类。如果图像是 Class A 的特定百分比和特定百分比，我如何制作模型 return Class B.
的百分比
这就是我想要的输出方式。
请帮忙。
由于它是二元分类器，我们将输出层的激活函数设置为 Sigmoid。在 sigmoid 中，结果值的范围从 0 到 1。如果结果是 Class A 的概率是 x，那么 Class B 的概率是 1-x。
因此代码将是，
pred = model.predict(img)
pred = pred.flatten().flatten()
classA_percent = float(pred)
classB_percent = float(1.0-pred)
pred_class = (pred > 0.5).astype(int)
我制作了一个 Tf/Keras CNN 模型，它是一个二元分类器。该模型对图像属于 Class A 还是 Class B 进行分类。如果图像是 Class A 的特定百分比和特定百分比，我如何制作模型 return Class B.
的百分比这就是我想要的输出方式。
请帮忙。
由于它是二元分类器，我们将输出层的激活函数设置为 Sigmoid。在 sigmoid 中，结果值的范围从 0 到 1。如果结果是 Class A 的概率是 x，那么 Class B 的概率是 1-x。
因此代码将是，
pred = model.predict(img)
pred = pred.flatten().flatten()
classA_percent = float(pred)
classB_percent = float(1.0-pred)
pred_class = (pred > 0.5).astype(int)