NVIDIA / earth2mip

Earth-2 Model Intercomparison Project (MIP) is a python framework that enables climate researchers and scientists to inter-compare AI models for weather and climate.
https://nvidia.github.io/earth2mip/
Apache License 2.0
187 stars 41 forks source link

Draft: Add statistics #138

Closed dallasfoster closed 7 months ago

dallasfoster commented 9 months ago

Earth-2 MIP Pull Request

Description

This PR will add a framework for statistics in earth2mip that will make it easier for users to

  1. Add statistics to an inference workflow.
  2. Add their own new statistics to their existing inference workflows.
  3. Create a statistics class that composes well with an IO pipeline.

Checklist

Dependencies

No new dependencies.