在Tensorflow2.6之前的版本中拥有predict_class属性
在结果预测时可以自动将结果变成0 1分布
y_train_predict = mlp.predict_classes(X_train)
print(y_train_predict)
输出结果为
而新版本的经上网查询只能用下面代码替代
y_train_predict = mlp.predict(X_train)
y_train_predict = np.argmax(y_train_predict, axis=1)
print(y_train_predict)
而打印出的预测结果确实这样
结果变成一维数据并且预测全为0,有什么办法可以解决?