Не удается получить доступ к индексу списка для максимального значения с argmax - PullRequest
1 голос
/ 27 марта 2019

Я работаю над наивным байесовским классификатором, и я получил этот список массивов в качестве предсказаний журнала:

predict = [array([-45.73329593, -47.23015876]), array([-56.83024746, -59.20630121]), array([-53.17297542, -55.19852072]), array([-35.67031864, -36.09391906]), array([-65.57891295, -66.81787995]), array([-62.64077968, -64.78048969]), array([-60.44866178, -61.96371683]), array([-46.36333681, -49.33896595]), array([-44.94102615, -46.89321275]), array([-58.67657099, -60.2740146 ]), array([-62.4623459 , -64.55626115])]

Я пытался запустить: np.argmax(predict, axis = 1), чтобы выбрать максимальное значениеpredict.Я использую argmax, чтобы вернуть соответствующий индекс, но он продолжает выдавать эту ошибку: TypeError: 'list' object is not callable.Почему выдает эту ошибку?

1 Ответ

0 голосов
/ 27 марта 2019

прогноз - list, а не numpy array

попытаться поместить список в массив numpy

>>> predict
[array([-45.73329593, -47.23015876]), array([-56.83024746, -59.20630121]), array([-53.17297542, -55.19852072]), array([-35.67031864, -36.09391906]), array([-65.57891295, -66.81787995]), array([-62.64077968, -64.78048969]), array([-60.44866178, -61.96371683]), array([-46.36333681, -49.33896595]), array([-44.94102615, -46.89321275]), array([-58.67657099, -60.2740146 ]), array([-62.4623459 , -64.55626115])]
>>> type(predict)
<class 'list'>
>>> pred_a = np.array(predict)
>>> type(pred_a)
<class 'numpy.ndarray'>
...