diff --git a/modules/deeplearning/cnn_cld_frac.py b/modules/deeplearning/cnn_cld_frac.py index 12f7147e6e01c73fabd98f2e32017f379e612bfe..15fd1dd6939d182b984f9ef15c0c9f3543f372da 100644 --- a/modules/deeplearning/cnn_cld_frac.py +++ b/modules/deeplearning/cnn_cld_frac.py @@ -228,9 +228,9 @@ def get_grid_cell_mean(grd_k): def get_label_data(grd_k): grd_k = np.where(np.isnan(grd_k), 0, grd_k) - cat_0 = np.logical_and(grd_k >= 0.0, grd_k < 0.2) - cat_1 = np.logical_and(grd_k >= 0.2, grd_k < 0.7) - cat_2 = np.logical_and(grd_k >= 0.7, grd_k <= 1.0) + cat_0 = np.logical_and(grd_k >= 0.0, grd_k < 0.15) + cat_1 = np.logical_and(grd_k >= 0.15, grd_k < 0.85) + cat_2 = np.logical_and(grd_k >= 0.85, grd_k <= 1.0) grd_k[cat_0] = -1 grd_k[cat_1] = 0 grd_k[cat_2] = 1