diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py index 8dceb4b977772c58fb7c6431b5f9b2a2c221939a..881aa0a8d6cabdb2b7a5219a22949a5ba512b44a 100644 --- a/modules/deeplearning/srcnn_l1b_l2.py +++ b/modules/deeplearning/srcnn_l1b_l2.py @@ -62,7 +62,7 @@ IMG_DEPTH = 1 label_param = 'cld_opd_dcomp' # label_param = 'cloud_probability' -params = ['temp_11_0um_nom', 'temp_12_0um_nom', 'refl_0_65um_nom', label_param] +params = ['temp_11_0um_nom', 'refl_0_65um_nom', label_param] data_params_half = ['temp_11_0um_nom'] data_params_full = ['refl_0_65um_nom'] @@ -665,10 +665,6 @@ class SRCNN: self.writer_valid.close() self.writer_train_valid_loss.close() - # 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() - def build_model(self): self.build_srcnn() self.model = tf.keras.Model(self.inputs, self.logits)