Numpy.argmax()

返回沿轴axis最大值的索引。索引

axis=0表明列,返回列的最大值索引位置

axis=1d表明行,返回行的最大值索引位置

主要用于图像分类,判断该图属于那种类别

相关文章
相关标签/搜索