在MXNet中如何确定标签的索引
我正在尝试理解python here中MXNet的手写数字识别。
创建训练数据和标签数据的代码如下所示：
def read_data(label_url, image_url):
with gzip.open(download_data(label_url)) as flbl:
magic, num = struct.unpack(">II", flbl.read(8))
label = np.fromstring(flbl.read(), dtype=np.int8)
with gzip.open(download_data(image_url), 'rb') as fimg:
magic, num, rows, cols = struct.unpack(">IIII", fimg.read(16))
image = np.fromstring(fimg.read(), dtype=np.uint8).reshape(len(label), rows, cols)
return (label, image)然后使用以下代码预测数字：
prob = model.predict(val_img[0:1].astype(np.float32)/255)[0]
assert max(prob) > 0.99, "Low prediction accuracy."
print 'Classified as %d with probability %f' % (prob.argmax(), max(prob))输出被分类为7，概率为0.999391。我的问题是MXNet是如何确定argmax函数返回的索引对应于label -7的
回答 1
Stack Overflow用户
发布于 2017-12-08 23:54:14
值7来自prob.argmax()。该方法返回得分最高的值的索引。
https://mxnet.incubator.apache.org/api/python/ndarray.html#mxnet.ndarray.argmax
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/42506907
复制相关文章
点击加载更多
相似问题
如何确定事件中的标签索引。
python中的mxnet ndarray索引
如何从mxnet nd数组中访问类标签的索引？输出必须是使用函数输入的标签名称的索引
在MXNet NDArray的列上使用索引数组
如何将softMax输出映射到MXNet中的标签
添加站长 进交流群
领取专属 10元无门槛券
AI混元助手 在线答疑
关注 腾讯云开发者公众号
洞察 腾讯核心技术
剖析业界实践案例
社区富文本编辑器全新改版！诚邀体验～
全新交互，全新视觉，新增快捷键、悬浮工具栏、高亮块等功能并同时优化现有功能，全面提升创作效率和体验
腾讯云开发者
