Open patel-zeel opened 1 year ago
Hi @patel-zeel, thank you for the suggestion. Because of travel, I will not be able to work on this for the next month. However, a feature like what you have suggested about saving latitude, longitude, and time is on the horizon for ClimateLearn. You can read more about my plans for the future of the library on this Google Doc. I plan to make a more formal post about these proposals when I return from my trip. In the meantime, I would very much appreciate you opening a PR to resolve this issue.
Is your feature request related to a problem? Please describe. The feature request is motivated by the following problems:
climate-learn
downscaling against my models and wanted to see if there is a pattern in errors, e.g., during certain hours of the day, predictions are more erroneous.trainer.test(model, dm)
provides a nice summary of metrics but does not save the predictions along withlat
,lon
,time
.Describe the solution you'd like A possible way could be to use a flow similar to
cl.utils.visualize_at_index
function.https://github.com/aditya-grover/climate-learn/blob/1a46b08e75ded7dde2d41867d01cfbe5d0d68c2c/src/climate_learn/utils/visualize.py#L10-L12
There could be a function
cl.utils.save_nc
which may look like:This function can save the predictions in exactly the same format as the data
nc
files withlat
,lon
andtime
co-ordinates. It should be able to retrievelat
,lon
,time
from data moduledm
.Willingness to work on a PR I'll be happy to work on a PR to make this happen!
Additional context This feature may also be useful to climate researchers who want to produce a time-lapse video of predictions with other libraries with additional geolayers similar to these examples in geemap library.