Update main.py

main.py

@@ -12,7 +12,7 @@ import random
 import argparse  # 명령행 인자를 파싱해주는 모듈
 import numpy as np
 from tqdm import tqdm  # 작업진행률 표시하는 라이브러리
-
+import csv
 import torch  # 파이토치
 import torch.nn as nn  # 클래스 # attribute 를 활용해 state 를 저장하고 활용
 import torch.nn.functional as F  # 함수 # 인스턴스화시킬 필요없이 사용 가능
@@ -154,8 +154,8 @@ def evaluate(model, dset_loader, criterion, is_print=False):
             #     'prediction': prediction,
             #     'confidence': confidence
             # })
-            with open("/home/user/app/result/ho.txt", 'w') as f:
-                f.writelines(prediction)
+            open("/home/user/app/result/ho.txt", 'a') as f
+            f.writelines(prediction+'\n')
 
             if is_print:
                 print()