从张量中获取最大索引

我的CNN模型通过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]

我想从这个NumPy数组中获取最大值的索引。我目前尝试使用以下代码(x是上面的数组):

result = x.index(max(x))

然而,这会引发一个错误,称这种数据类型不支持.index方法?


回答:

你可以简单地使用np.argmax函数:

import numpy as nppreds = model.predict(test_data)pred_class = np.argmax(preds, axis=-1)

Related Posts

在使用k近邻算法时,有没有办法获取被使用的“邻居”?

我想找到一种方法来确定在我的knn算法中实际使用了哪些…

Theano在Google Colab上无法启用GPU支持

我在尝试使用Theano库训练一个模型。由于我的电脑内…

准确性评分似乎有误

这里是代码: from sklearn.metrics…

Keras Functional API: “错误检查输入时:期望input_1具有4个维度,但得到形状为(X, Y)的数组”

我在尝试使用Keras的fit_generator来训…

如何使用sklearn.datasets.make_classification在指定范围内生成合成数据?

我想为分类问题创建合成数据。我使用了sklearn.d…

如何处理预测时不在训练集中的标签

已关闭。 此问题与编程或软件开发无关。目前不接受回答。…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注