torch_timeseries.dataset

Forecast/Imputation Datasets

Traffic

Traffic dataset for road occupancy rates in the San Francisco Bay area.

ExchangeRate

The collection of the daily exchange rates of eight foreign countries including Australia, British, Canada, Switzerland, China, Japan, New Zealand, and Singapore ranging from 1990 to 2016.

SolarEnergy

The dataset contains solar power production records for the year 2006, sampled every 5 minutes from 137 PV plants in Alabama State.

Electricity

The raw dataset is available at https://archive.ics.uci.edu/ml/datasets/ElectricityLoadDiagrams20112014.

ETTh1

Electricity Transformer Temperature hourly.

ETTh2

Electricity Transformer Temperature hourly.

ETTm1

Electricity Transformer Temperature minutely~(15 minutes level).

ETTm2

Electricity Transformer Temperature minutely~(15 minutes level).

Weather

This dataset contains local climatological data for nearly 1,600 in US locations, covering 4 years from 2010 to 2013.

ILI

ILI (Influenza-Like Illness) records the weekly ratio of influenza-like illness patients versus the total patients by the Centers for Disease Control and Prevention (CDC) of the United States from 2002 to 2021.

M4

The M4 forecasting competition, the continuation of the previous three ones organized by Spyros Makridakis (https://en.wikipedia.org/wiki/Makridakis_Competitions).

Classification Datasets

UEA

Dataset class for datasets included in: Time Series Classification Archive (www.timeseriesclassification.com).

AnomalyDetection Datasets

SWaT

SWaT (Safe Water Treatment, Mathur & Tippenhauer (2016)) is obtained from 51 sensors in continuously operating critical infrastructure systems.

SMD

SMD (Server Machine Dataset, Su et al. (2019)) is a 5-week dataset collected from a large Internet company with 38 dimensions.

SMAP

SMAP is a multidimensional time series dataset collected by NASA's Mars rover.

MSL

MSL is a public dataset from NASA with 55 dimensions showing the status of sensor and actuator data from the Mars Rover.

PSM

PSM is a public dataset from eBay's internal server nodes.

Synthetic Datasets

Dummy

Dummy dataset for testing purposes.

DummyGraph

Dummy graph dataset for testing purposes.

SimFreqCF

param root:

data save location, defaults to ~/.torchtimeseries/data

SimFreq

param root:

data save location, defaults to ~/.torchtimeseries/data