问题:
1、使用keras做分类任务
2、model.predict预测得到的值为每个类别的概率值,而不是类别
3、y_test_pred = model.predict(x_test, batch_size=256, verbose=1)
解决:
import numpy as np
y_test_pred = np.argmax(y_test_pred, axis=1)
注意:axis值为你要转换的行,而且输出为一维数据,即每个样本对应一个类别值
问题:
1、使用keras做分类任务
2、model.predict预测得到的值为每个类别的概率值,而不是类别
3、y_test_pred = model.predict(x_test, batch_size=256, verbose=1)
解决:
import numpy as np
y_test_pred = np.argmax(y_test_pred, axis=1)
注意:axis值为你要转换的行,而且输出为一维数据,即每个样本对应一个类别值