From ca415832f1efd50407302a2fb379f7ce78ba0a48 Mon Sep 17 00:00:00 2001 From: lpalma Date: Thu, 27 Jun 2024 17:48:46 +0200 Subject: [PATCH 1/2] fixes in metrics init --- tinysunset/verification/__init__.py | 3 ++- tinysunset/verification/metrics.py | 2 ++ 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/tinysunset/verification/__init__.py b/tinysunset/verification/__init__.py index 0831cd1..ff30577 100644 --- a/tinysunset/verification/__init__.py +++ b/tinysunset/verification/__init__.py @@ -5,4 +5,5 @@ logger = logging.getLogger(__name__) from .bootrstrap import _xbootstrap from .metrics import comp_metrics, plot_metric, skill_scores -from .verification import Verification, VerificationConfig \ No newline at end of file +from .verification import Verification, VerificationConfig, parse_verification_config +from .scorecards import compute_scorecard, plot_scorecard \ No newline at end of file diff --git a/tinysunset/verification/metrics.py b/tinysunset/verification/metrics.py index 78eb864..6231f0c 100644 --- a/tinysunset/verification/metrics.py +++ b/tinysunset/verification/metrics.py @@ -188,6 +188,7 @@ def plot_metric( "crpss": dict(cmap=_crpss_cmap, vmin=0, vmax=1, levels=21), "rpss": dict(cmap=_crpss_cmap, vmin=0, vmax=1, levels=21), + "bss": dict(cmap=_crpss_cmap, vmin=0, vmax=1, levels=21), "mses": dict(cmap=_crpss_cmap, vmin=0, vmax=1, levels=21), "rmses": dict(cmap=_crpss_cmap, vmin=0, vmax=1, levels=21), "maes": dict(cmap="RdBu_r", vmin=-1, vmax=1, levels=21), @@ -212,6 +213,7 @@ def plot_metric( "crpss": "CRPSS", "rpss": "RPSS", + "bss": "BSS", "auprc_neg1sig": "AUPRC [-1$\sigma$]", "auprc_1sig": "AUPRC [1$\sigma$]", "auprc_neg2sig": "AUPRC [-2$\sigma$]", -- GitLab From b8985c67e5e647584abe1b9be8bb34786b0c7bf4 Mon Sep 17 00:00:00 2001 From: lpalma Date: Mon, 1 Jul 2024 15:30:55 +0200 Subject: [PATCH 2/2] dockerfile --- Dockerfile | 25 +++++++++++++++++++++++++ 1 file changed, 25 insertions(+) create mode 100644 Dockerfile diff --git a/Dockerfile b/Dockerfile new file mode 100644 index 0000000..bac6b24 --- /dev/null +++ b/Dockerfile @@ -0,0 +1,25 @@ +FROM nvcr.io/nvidia/pytorch:23.10-py3 +RUN pip install zarr +RUN pip install \ +cartopy \ +chex \ +dask \ +dm-haiku \ +dm-tree \ +matplotlib \ +numpy \ +pandas \ +rtree \ +scipy \ +trimesh \ +typing_extensions \ +xarray \ +ipython \ +ipywidgets \ +seaborn \ +xclim \ +ipywidgets \ +netCDF4 \ +cartopy \ +xskillscore \ +jupyterlab -- GitLab