Update DLStress.py

marcin-laskowski · Jun 13, 2018 · 734a797 · 734a797
1 parent b36a9ae
commit 734a797
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/DLStress.py b/DLStress.py
@@ -359,7 +359,7 @@ def CreateModel_StressDecoding(W1_in, W2_in):
 # %%
 # to save the predicted stress distribution during cross validation
 # insert the function below to the code block of the cross validation
-# sio.savemat('StressData_pred.mat', {'Sp':Sp, 'idx_test':idx_test})
+sio.savemat('StressData_pred.mat', {'Sp':Sp, 'idx_test':idx_test})
 
 
 # %% show the time cost on the testing set: input shape, output stress