用平台提供的代码,在平台跑程序,测试一张图片时,怎么修改才能得到每种分类的概率,而不是直接给出这是什么类?
顶楼上
能养成提问贴代码的好习惯么……简单来说就是prediction = argmax(logits), 只要把logits,就是网络最后一层的输出(softmax(wx + b))的结果,打出来就可以了
域名投资