如何将预测结果循环写入一个csv文件中

Question

如何将预测结果循环写入一个csv文件中，将每一个Y[j]写入到同一个csv文件中

Y= []
tmp_x1 = X_train.copy()
tmp_x2 = X_test.copy()
for j in range(100):  
    for i in range(6585): 
        tmp_x1['X1'][i] = X1[j]
        tmp_Y1 = list(M.predict(tmp_x1))
    for i in range(1966): 
        tmp_x2['X1'][i] = X1[j]
        tmp_Y2 = list(M.predict(tmp_x2))
    Y.append(tmp_Y1 + tmp_Y2)

Answer 1

可以尝试用该链接中的方式，去实现：https://blog.csdn.net/qq_41915623/article/details/125834267

    os.makedirs(os.path.join('..', 'data'), exist_ok=True)  # 创建数据文件夹
    data_file = os.path.join('..', 'data', 'loss.csv')
    with open(data_file, 'w', encoding='utf-8', newline='') as f:
        writer = csv.writer(f)
        writer.writerow(['epoch', 'batch_index', 'loss'])