snapshot...

modules/deeplearning/cloud_fraction_fcn.py

@@ -8,6 +8,7 @@ import os, datetime
 import numpy as np
 import pickle
 import h5py
+import xarray as xr
 import gc
 
 LOG_DEVICE_PLACEMENT = False
@@ -816,6 +817,15 @@ def run_evaluate_static(in_file, out_file, ckpt_dir):
     bt = denormalize(bt, 'temp_11_0um_nom', mean_std_dct)
     refl_avg = denormalize(refl_avg, 'refl_0_65um_nom', mean_std_dct)
 
+    var_names = ['cloud_fraction', 'temp_11_0um', 'refl_0_65um']
+    dims = ['num_params', 'y', 'x']
+
+    da = xr.DataArray(np.column_stack([cld_frac_out, bt, refl_avg]), dims=dims)
+    da.assign_coords({
+        'lat': (['y', 'x'], lats),
+        'lon': (['y', 'x'], lons)
+    })
+
     if out_file is not None:
         np.save(out_file, (cld_frac_out, bt, refl_avg, cp, lons, lats))
     else: