keras predict_生成器批量保存预测

我有一组我想预测的文件。我已经编写了一个生成器函数，它读取文件并创建用于输入的数组来预测生成函数。当前，prediction_generator函数返回一个数组，其中包含所有文件的预测。是我保存/访问各个批次预测的一种方式，因为我以后需要将它们与原始文件合并

def generate_prediction_array(pred_files):
    for file in pred_files:
        X = create_array(file)
        yield X
...
y_pred_prob = model.predict_generator(generate_prediction_array(pred_files), 
                                      max_queue_size=2,
                                      steps=len(pred_array_files))

---

当我看y_pred_prob时，它对所有文件中的所有数据点都有预测。相反，我想分别访问每个批次的预测，以便将其与原始文件数据合并。

如何使用

for

循环而不是使用预测生成器，将所有结果存储到单独的文件中，然后再合并它们？您能详细解释一下吗