diff --git a/Dockerfile b/Dockerfile new file mode 100644 index 0000000000000000000000000000000000000000..bac6b246b7dabb05d8284292b2b340d12cab9f33 --- /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 diff --git a/tinysunset/verification/__init__.py b/tinysunset/verification/__init__.py index 0831cd1a3a008294ea97685c6bb17f28117e4c76..ff30577b9edb8bce7e91fb85e28f1cdad387ca56 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 78eb8647582d3b650763711aa5a0030e69427b93..6231f0c7faaca1096c61301b41d410dd351e5f3a 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$]",