Skip to main content

Nowcasting Dataset

Project description

nowcasting_dataset

A multi-process data loader for PyTorch which aligns three separate datasets:

  • Satellite imagery (EUMETSAT SEVIRI RSS 5-minutely data of UK)
  • Numerical Weather Predictions (NWPs. UK Met Office UKV model from CEDA)
  • Solar PV power timeseries data (from PVOutput.org, downloaded using our pvoutput Python code.)

When we first started writing nowcasting_dataset, our intention was to load and align data from these three datasets on-the-fly during ML training. And nowcasting_dataset can still be used that way! But it just isn't quite fast enough to keep a modern GPU constantly fed with data when loading multiple satellite channels and multiple NWP parameters. So, now, this code is used to pre-prepare thousands of batches, and save these batches to disk, each as a separate NetCDF file. These files can then be loaded super-quickly at training time. The end result is a 12x speedup in training.

The script scripts/prepare_ml_data.py is used to pre-compute the training and validation data (the script makes use of the nowcasting_dataset library). nowcasting_dataset.dataset.datasets.NetCDFDataset is a PyTorch Dataset which loads the pre-prepared batches during ML training.

This repo doesn't contain the ML models themselves. The models are in: https://github.com/openclimatefix/predict_pv_yield/ and https://github.com/openclimatefix/satflow, and utils are in https://github.com/openclimatefix/nowcasting_utils

Please see this page for an overview of the Open Climate Fix solar PV nowcasting project, and how our code repositories fit together.

Installation

Conda

From within the cloned nowcasting_dataset directory:

conda env create -f environment.yml
conda activate nowcasting_dataset
pip install -e .

Note you can install the pytorch and pytorch_lightning using

pip install -e .[torch]

but it is only used to create a dataloader for machine learning models.

Pip

A (probably older) version is also available through pip install nowcasting-dataset

RuntimeError: unable to open shared memory object

To prevent PyTorch failing with an error like RuntimeError: unable to open shared memory object </torch_2276740_2849291446> in read-write mode, edit /etc/security/limits.conf as root and add this line: * soft nofile 512000 then log out and log back in again (see this issue for more details).

PV Live API

If you want to also install PVLive then use pip install git+https://github.com/SheffieldSolar/PV_Live-API

Pre-commit

A pre commit hook has been installed which makes black run with every commit. You need to install black and pre-commit (these will be installed by conda or pip when installing nowcasting_dataset) and run pre-commit install in this repo.

Testing

To test using the small amount of data stored in this repo: py.test -s

To test using the full dataset on Google Cloud, add the --use_cloud_data switch.

Documentation

Please see the Example class for documentation about the different data fields in each example / batch.

Project details


Release history Release notifications | RSS feed

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

nowcasting_dataset-0.2.1.tar.gz (70.0 kB view hashes)

Uploaded Source

Built Distribution

nowcasting_dataset-0.2.1-py3-none-any.whl (92.9 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page