Numpy.argmax()

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

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

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

主要用於圖像分類,判斷該圖屬於那種類別

相關文章
相關標籤/搜索