Skip to content
Snippets Groups Projects
Commit a965e4f5 authored by tomrink's avatar tomrink
Browse files

minor...

parent c690c697
No related branches found
No related tags found
No related merge requests found
import glob import glob
import tensorflow as tf import tensorflow as tf
from util.setup import logdir, modeldir, cachepath, now, ancillary_path from util.setup import logdir, modeldir, cachepath, now, ancillary_path, home_dir
from util.util import EarlyStop, normalize, make_for_full_domain_predict from util.util import EarlyStop, normalize, make_for_full_domain_predict
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
...@@ -925,7 +925,7 @@ class UNET: ...@@ -925,7 +925,7 @@ class UNET:
if self.h5f_l2_tst is not None: if self.h5f_l2_tst is not None:
self.h5f_l2_tst.close() self.h5f_l2_tst.close()
f = open('/home/rink/best_stats_'+now+'.pkl', 'wb') f = open(home_dir+'/best_stats_'+now+'.pkl', 'wb')
pickle.dump((best_test_loss, best_test_acc, best_test_recall, best_test_precision, best_test_auc, best_test_f1, best_test_mcc), f) pickle.dump((best_test_loss, best_test_acc, best_test_recall, best_test_precision, best_test_auc, best_test_f1, best_test_mcc), f)
f.close() f.close()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment