进行keras预测后的嵌套列表

Question

我使用以下方法根据我的 keras 模型进行预测：

# fit the keras model on the dataset
ker_model.fit(xtrain, ytrain, epochs=200, verbose=0)    
predictions = ker_model.predict(xtest)
predictions = predictions.astype(int)
predictions.mean()
predictions

但是，问题是，我的预测是在嵌套数组列表中。意思如下：

array([[0],
       [0],
       [0],
       [1],
       [1]])

我如何才能确保我的预测最终不会出现在这样的嵌套列表中，或者取消列出预测？

我希望我的输出看起来像：

array([0, 0, 0, 1, 1])

Answer 1

你可以使用 predictions =predictions.ravel() 要么 predictions =predictions.squeeze()

进行keras预测后的嵌套列表

Nested list after making a keras prediction

python

arrays

classification

keras