paddlehub 推荐返回概率
收藏
paddlehub1.8 版本 MultiLabelClassif.predict(data=encoded_data, label_list=label_list) 怎么返回每个标签对应的概率 目前返回的是 标签对应0和1 怎么返回概率呢
1
收藏
请登录后评论
分类的结果会返回各个类别的概率,然后选最大的类别作为最后输出的标签
最后一层经过softmax激活后输出的就是各个类别的概率
比如最后一层全连接层形状是102,那么输出的就是这102个类别,每个类别的概率
会报错啊
predict() got an unexpected keyword argument 'return_prob'