开发者社区> 问答> 正文

PAI示例cifar10 分类问题

用平台提供的代码,在平台跑程序,测试一张图片时,怎么修改才能得到每种分类的概率,而不是直接给出这是什么类?

展开
收起
Chin2018 2018-03-19 19:32:18 2124 0
3 条回答
写回答
取消 提交回答
  • 顶楼上

    2019-07-17 22:02:19
    赞同 展开评论 打赏
  • 能养成提问贴代码的好习惯么……
    简单来说就是prediction = argmax(logits), 只要把logits,就是网络最后一层的输出(softmax(wx + b))的结果,打出来就可以了

    2019-07-17 22:02:19
    赞同 展开评论 打赏
  • 域名投资

    2019-07-17 22:02:19
    赞同 1 展开评论 打赏
问答排行榜
最热
最新

相关电子书

更多
大规模机器学习在蚂蚁+阿里的应用 立即下载
阿里巴巴机器学习平台AI 立即下载
微博机器学习平台架构和实践 立即下载