进行测试 with torch.no_grad(): # 测试模式 model.eval() # 计算测试集上的损失 test_loss = 0 for i in range(len(input_test)): inputs = input_test[i] outputs = model(inputs) labels = output_test[i] # 计算损失 loss = criterion(outputs, labels) test_loss += loss.item() # 输出模拟值outputs print(f'Output{i}', outputs) print('Test Loss: %.4f' % (test_loss / len(input_test))) 对于这段代码,其中的inputs,outputs,labels都是6 * 32 * 32的三维数组,我想将每一步的inputs, outputs,labels按照顺序整合到一个excel文件中输出,其中inputs都放在sheet1,outputs都放在sheet2,labels都放在sheet3,每一个inputs,outputs,labels都有一个编号,例如第一个inputs叫做test_input_1,第一个outputs叫做test_output_1,第一个labels叫做test_label_1,以此类推
时间: 2023-06-18 18:03:22 浏览: 63
pytorch_test_intro:在pytorch中测试简介
可以使用Python中的openpyxl库来创建和操作Excel文件。首先,需要安装openpyxl库,可以使用以下命令进行安装:
```
pip install openpyxl
```
然后,可以按照以下方式修改您的代码来将数据写入Excel文件:
```
import openpyxl
# 创建Excel文件
workbook = openpyxl.Workbook()
# 创建sheet1
sheet1 = workbook.create_sheet("inputs", 0)
for i in range(len(input_test)):
# 写入数据
for x in range(6):
for y in range(32):
for z in range(32):
sheet1.cell(row=x*32+y+1, column=z+1, value=inputs[x, y, z])
# 重命名sheet
sheet1.title = "test_input_" + str(i+1)
# 创建sheet2
sheet2 = workbook.create_sheet("outputs", 1)
for i in range(len(output_test)):
# 写入数据
for x in range(6):
for y in range(32):
for z in range(32):
sheet2.cell(row=x*32+y+1, column=z+1, value=outputs[x, y, z])
# 重命名sheet
sheet2.title = "test_output_" + str(i+1)
# 创建sheet3
sheet3 = workbook.create_sheet("labels", 2)
for i in range(len(output_test)):
# 写入数据
for x in range(6):
for y in range(32):
for z in range(32):
sheet3.cell(row=x*32+y+1, column=z+1, value=labels[x, y, z])
# 重命名sheet
sheet3.title = "test_label_" + str(i+1)
# 保存Excel文件
workbook.save("output.xlsx")
```
在这段代码中,我们使用openpyxl库创建了一个新的Excel文件,并在其中创建了三个工作表(sheet1,sheet2和sheet3)。然后,我们使用两个嵌套的for循环将每个数据点中的数据写入单元格。最后,我们使用`save()`函数将Excel文件保存到磁盘上。
阅读全文