feat: add Italy Air Quality and release v0.4;

README.md

@@ -46,7 +46,7 @@
     </a>
 </p>
 
-> 📣 TSDB now supports a total of 1️⃣6️⃣9️⃣ time-series datasets ‼️
+> 📣 TSDB now supports a total of 1️⃣7️⃣0️⃣ time-series datasets ‼️
 
 <a href='https://github.com/WenjieDu/PyPOTS'><img src='https://pypots.com/figs/pypots_logos/PyPOTS/logo_FFBG.svg' width='160' align='left' /></a>
 TSDB is a part of
@@ -108,6 +108,7 @@ That's all. Simple and efficient. Enjoy it! 😃
 | [PhysioNet Challenge 2012](dataset_profiles/physionet_2012)                                       | Forecasting, Imputation, Classification |
 | [PhysioNet Challenge 2019](dataset_profiles/physionet_2019)                                       | Forecasting, Imputation, Classification |
 | [Beijing Multi-Site Air-Quality](dataset_profiles/beijing_multisite_air_quality)                  | Forecasting, Imputation                 |
+| [Italy Air Quality](dataset_profiles/italy_air_quality)                                           | Forecasting, Imputation                 |
 | [Electricity Load Diagrams](dataset_profiles/electricity_load_diagrams)                           | Forecasting, Imputation                 |
 | [Electricity Transformer Temperature (ETT)](dataset_profiles/electricity_transformer_temperature) | Forecasting, Imputation                 |
 | [Vessel AIS](dataset_profiles/vessel_ais)                                                         | Forecasting, Imputation, Classification |
@@ -121,7 +122,7 @@ and we are pursuing to publish it in prestigious academic venues, e.g. JMLR (tra
 please cite PyPOTS project as below and 🌟star this repository to make others notice this library. 🤗 Thank you!
 
 <p align="center">
-<a href="https://pypots.com/ecosystem/">
+<a href="https://github.com/WenjieDu/PyPOTS">
     <img src="https://pypots.com/figs/pypots_logos/Ecosystem/PyPOTS_Ecosystem_Pipeline.png" width="95%"/>
 </a>
 </p>

dataset_profiles/italy_air_quality/README.md

@@ -0,0 +1,17 @@
+# Italy Air Quality
+
+## Citing this dataset 🤗
+
+`Vito,Saverio. (2016). Air Quality. UCI Machine Learning Repository. https://doi.org/10.24432/C59K5F`
+
+or
+
+```bibtex
+@misc{vito2016air,
+author       = {Vito,Saverio},
+title        = {{Air Quality}},
+year         = {2016},
+howpublished = {UCI Machine Learning Repository},
+note         = {{DOI}: https://doi.org/10.24432/C59K5F}
+}
+```

tsdb/__init__.py

@@ -21,7 +21,7 @@
 #
 # Dev branch marker is: 'X.Y.dev' or 'X.Y.devN' where N is an integer.
 # 'X.Y.dev0' is the canonical version of 'X.Y.dev'
-__version__ = "0.3.1"
+__version__ = "0.4"
 
 from .data_processing import (
     CACHED_DATASET_DIR,

tsdb/data_processing.py

@@ -18,6 +18,7 @@
     load_beijing_air_quality,
     load_ucr_uea_dataset,
     load_ais,
+    load_italy_air_quality,
 )
 from .utils.downloading import download_and_extract
 from .utils.file import purge_path, pickle_load, pickle_dump, determine_data_home
@@ -100,6 +101,8 @@ def load(dataset_name: str, use_cache: bool = True) -> dict:
                 result = load_ett(dataset_saving_path)
             elif dataset_name == "beijing_multisite_air_quality":
                 result = load_beijing_air_quality(dataset_saving_path)
+            elif dataset_name == "italy_air_quality":
+                result = load_italy_air_quality(dataset_saving_path)
             elif dataset_name == "vessel_ais":
                 result = load_ais(dataset_saving_path)
             elif "ucr_uea_" in dataset_name:

tsdb/database.py

@@ -30,6 +30,8 @@
     # https://github.com/WenjieDu/TSDB/tree/main/dataset_profiles/beijing_multisite_air_quality
     "beijing_multisite_air_quality": "https://archive.ics.uci.edu/ml/machine-learning-databases/00501/"
     "PRSA2017_Data_20130301-20170228.zip",
+    # https://github.com/WenjieDu/TSDB/tree/main/dataset_profiles/italy_air_quality
+    "italy_air_quality": "https://archive.ics.uci.edu/static/public/360/air+quality.zip",
     #
     # https://github.com/WenjieDu/TSDB/tree/main/dataset_profiles/vessel_ais
     "vessel_ais": "https://zenodo.org/record/8064564/files/parquets.zip",

tsdb/loading_funcs/__init__.py

@@ -7,11 +7,12 @@
 
 from .beijing_multisite_air_quality import load_beijing_air_quality
 from .electricity_load_diagrams import load_electricity
+from .electricity_transformer_temperature import load_ett
+from .italy_air_quality import load_italy_air_quality
 from .physionet_2012 import load_physionet2012
 from .physionet_2019 import load_physionet2019
 from .ucr_uea_datasets import load_ucr_uea_dataset
 from .vessel_ais import load_ais
-from .electricity_transformer_temperature import load_ett
 
 __all__ = [
     "load_beijing_air_quality",
@@ -21,4 +22,5 @@
     "load_ucr_uea_dataset",
     "load_ais",
     "load_ett",
+    "load_italy_air_quality",
 ]

tsdb/loading_funcs/italy_air_quality.py

@@ -0,0 +1,39 @@
+"""
+Scripts related to dataset Italy Air Quality.
+
+For more information please refer to:
+https://github.com/WenjieDu/TSDB/tree/main/dataset_profiles/italy_air_quality
+"""
+
+# Created by Wenjie Du <[email protected]>
+# License: BSD-3-Clause
+
+import os
+
+import pandas as pd
+
+
+def load_italy_air_quality(local_path):
+    """Load dataset Italy Air Quality.
+
+    Parameters
+    ----------
+    local_path : str,
+        The local path of dir saving the raw data of Beijing Multi-site Air Quality.
+
+    Returns
+    -------
+    data : dict
+        A dictionary contains X:
+            X : pandas.DataFrame
+                The time-series data of Beijing Multi-site Air Quality.
+    """
+    file_path = os.path.join(local_path, "AirQualityUCI.csv")
+    df = pd.read_csv(file_path, sep=";", decimal=",")
+    # remove empty columns
+    df.drop(columns=["Unnamed: 15", "Unnamed: 16"], inplace=True)
+
+    data = {
+        "X": df,
+    }
+    return data