update

utils.py

@@ -106,8 +106,7 @@ def glue_data(file_name, total, output):
 def save_data(data, filename):
     with open(filename, 'w', newline='') as csvfile:
         writer = csv.writer(csvfile)
-        for row in data:
-            writer.writerow(row)
+        writer.writerows(data)
 
 def dataDelete(path):
     try:
@@ -223,10 +222,12 @@ def reconstruct(model_name, total, filepath, outputfile):
 
         # Deep Learning Artifact Removal
         d_data = decode_data(data_noise, std, model_name)
-        d_data = d_data[0]
+        #d_data = d_data[0]
 
         outputname = filepath + '/temp2/output{}.csv'.format(str(i))
         save_data(d_data, outputname)
+        #d_data.to_csv(outputname, index=False)
+
 
     # --------------------glue_data----------------------------
     glue_data(filepath+"/temp2/", total, filepath+'/'+outputfile)