From a965e4f5d8e8bdc579ec3234806798a983dc7b90 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Thu, 14 Apr 2022 15:06:10 -0500
Subject: [PATCH] minor...
---
modules/deeplearning/unet.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/modules/deeplearning/unet.py b/modules/deeplearning/unet.py
index 3d7961de..2f3ae5f1 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()
--
GitLab