开发者社区> 问答> 正文

PAI示例cifar10 分类问题

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

展开
收起
Chin2018 2018-03-19 19:32:18 2132 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 展开评论 打赏
问答排行榜
最热
最新

相关电子书

更多
大规模机器学习在蚂蚁+阿里的应用 立即下载
基于Spark的面向十亿级别特征的 大规模机器学习 立即下载
基于Spark的大规模机器学习在微博的应用 立即下载