我的 CNN 產(chǎn)生了以下內(nèi)容（來(lái)自model.predict()）：Tensor("input_1:0", shape=(?, 2, 26, 1), dtype=float32)[9.9952221e-01 2.3613637e-04 1.9953270e-06 1.6922619e-05 2.2012556e-04 2.4441533e-07 3.5276526e-07 7.4913805e-07 4.0657511e-07 8.7760031e-07]我想從這個(gè) numpy 數(shù)組中獲取最大值的索引?，F(xiàn)在，我已經(jīng)嘗試過(guò)這樣做（x即上面的數(shù)組）：result = x.index(max(x))相反，這會(huì)引發(fā)一個(gè)錯(cuò)誤，指出此數(shù)據(jù)類(lèi)型不支持.index?
1 回答
三國(guó)紛爭(zhēng)
TA貢獻(xiàn)1804條經(jīng)驗(yàn) 獲得超7個(gè)贊
您可以簡(jiǎn)單地使用該np.argmax功能：
import numpy as np
preds = model.predict(test_data)
pred_class = np.argmax(preds, axis=-1)
添加回答
舉報(bào)
0/150
提交
取消