分享

model.predict()未生成预期的标签?

 行走在理想边缘 2022-05-24 发布于四川

种解决方案是使用简单的统计解释,其中我们将使用0.5截止值。因此,高于0.5的所有内容将被视为1,低于0的内容将被视为0

import numpy as np

pred =  np.array([[0.9434484 ]
 ,[0.3787447 ]
 ,[0.87870705]
 ,[0.7575223 ]
 ,[0.39714795]])

np.round(pred)
Out[37]: 
array([[1.],
       [0.],
       [1.],
       [1.],
       [0.]])

如果结果不是一个概率,那么:

def sigmoid(x):
  return 1 / (1 + math.exp(-x))

必须用于将其缩放到0-1比例

步骤5 model.predict(x_test)可以替换为:

model.predict_classes(x_test)

在序列模型中预测类。如果您将来在功能模型中需要此功能,以下是解决方案:

y_prob = model.predict(x_test) 
y_classes = y_prob.argmax(axis=-1)

    本站是提供个人知识管理的网络存储空间,所有内容均由用户发布,不代表本站观点。请注意甄别内容中的联系方式、诱导购买等信息,谨防诈骗。如发现有害或侵权内容,请点击一键举报。
    转藏 分享 献花(0

    0条评论

    发表

    请遵守用户 评论公约

    类似文章 更多