diff --git a/modules/deeplearning/unet.py b/modules/deeplearning/unet.py index 3d7961deac188df86d2b705f4183ff07164894b6..2f3ae5f1ef01bb5c4f39c601028b1d406a42af14 100644 --- a/modules/deeplearning/unet.py +++ b/modules/deeplearning/unet.py @@ -1,6 +1,6 @@ import glob 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 import matplotlib.pyplot as plt @@ -925,7 +925,7 @@ class UNET: if self.h5f_l2_tst is not None: 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) f.close()