python - 如何从 Keras 对 model.predict 函数的输出进行排序？

Question

我已经建立了一个序列模型来预测数据集。输出是一个值数组，如下所示：

[[25.672846 ]
 [35.517834 ]
 [16.25976  ]
 [27.06339  ]
 [15.176579 ]
 [20.261982 ]
 [17.346558 ]
 [14.3797455]
 [23.327494 ]
 [17.967245 ]
 [17.85637  ]
 [16.672417 ]
 [ 6.2621603]
 [20.35404  ]]

我想按升序对数组进行排序。

我试过了

pred = np.sort(model.predict(data))

和

pred = np.argsort(model.predict(data))

我失败了。

标签： pythonnumpy

Answer 1

你应该使用np.sort如下：

pred = np.sort(model.predict(data), axis=0)

默认行为axis=-1相当于axis=1您的情况。