Add time-series datasets

synthesized-io · Aug 19, 2022 · 52cee6b · 52cee6b
1 parent 92fadbf
commit 52cee6b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/src/synthesized_datasets/__init__.py b/src/synthesized_datasets/__init__.py
@@ -13,4 +13,10 @@
 _Dataset("telecom_churn", "tabular/templates/telecom-churn.csv", [_Tag.CHURN, _Tag.BINARY_CLASSIFICATION])
 _Dataset("vehicle_insurance", "tabular/templates/vehicle-insurance.csv", [_Tag.INSURANCE])
 
+_Dataset("air_quality", "time-series/air-quality.csv", [_Tag.TIME_SERIES])
+_Dataset("bitcoin_price", "time-series/bitcoin_price.csv", [_Tag.FINANCE, _Tag.TIME_SERIES])
+_Dataset("brent_oil_prices", "time-series/brent-oil-prices.csv", [_Tag.FINANCE, _Tag.TIME_SERIES])
+_Dataset("occupancy_data", "time-series/occupancy-data.csv", [_Tag.TIME_SERIES])
+_Dataset("s&p500_5yr", "time-series/sandp500_5yr.csv", [_Tag.FINANCE, _Tag.TIME_SERIES])
+
 from ._datasets import *
diff --git a/src/synthesized_datasets/_datasets.py b/src/synthesized_datasets/_datasets.py
@@ -15,6 +15,8 @@ class _Tag(_Enum):
     BINARY_CLASSIFICATION = "binary_classification"
     CHURN = "churn"
     REGRESSION = "regression"
+    TIME_SERIES = "time_series"
+    FINANCE = "finance"
 
     def __repr__(self):
         return f"<{self.value}>"