np.argmax(predictions,axis=1)
时间: 2023-06-13 20:07:00 浏览: 251
详解numpy的argmax的具体使用
这段代码使用numpy的argmax函数来获取预测结果中每个样本最大概率对应的类别标签。其中,predictions是一个二维数组,每行表示一个样本的概率分布,每列表示一个类别的概率。axis=1表示按行进行操作,即对每个样本的概率分布进行操作。最终返回的是一个一维数组,包含了每个样本最大概率对应的类别标签。
阅读全文
详解numpy的argmax的具体使用