Keras model.predict输出的概率值转换为类别

问题：
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值为你要转换的行，而且输出为一维数据，即每个样本对应一个类别值