Skip to content

[ENH] AutoETS, ARIMA, NaiveForecaster, Dataset loading updates, Differencing/Windowing/Train test transforms. #2828

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
wants to merge 12 commits into
base: main
Choose a base branch
from
11 changes: 10 additions & 1 deletion aeon/datasets/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,10 @@
"load_human_activity_segmentation_datasets",
# Write functions
"write_to_ts_file",
"write_to_tsf_file",
"write_to_arff_file",
"write_regression_dataset",
"write_forecasting_dataset",
# Single problem loaders
"load_airline",
"load_arrow_head",
Expand Down Expand Up @@ -57,7 +60,13 @@
load_from_tsv_file,
load_regression,
)
from aeon.datasets._data_writers import write_to_arff_file, write_to_ts_file
from aeon.datasets._data_writers import (
write_forecasting_dataset,
write_regression_dataset,
write_to_arff_file,
write_to_ts_file,
write_to_tsf_file,
)
from aeon.datasets._single_problem_loaders import (
load_acsf1,
load_airline,
Expand Down
6 changes: 3 additions & 3 deletions aeon/datasets/_data_loaders.py
Original file line number Diff line number Diff line change
Expand Up @@ -977,7 +977,7 @@ def load_forecasting(name, extract_path=None, return_metadata=False):
>>> X=load_forecasting("m1_yearly_dataset") # doctest: +SKIP
"""
# Allow user to have non standard extract path
from aeon.datasets.tsf_datasets import tsf_all
from aeon.datasets.tsf_datasets import tsf_monash

if extract_path is not None:
local_module = extract_path
Expand All @@ -993,8 +993,8 @@ def load_forecasting(name, extract_path=None, return_metadata=False):
if name not in get_downloaded_tsf_datasets(extract_path):
# Dataset is not already present in the datasets directory provided.
# If it is not there, download and install it.
if name in tsf_all.keys():
id = tsf_all[name]
if name in tsf_monash.keys():
id = tsf_monash[name]
if extract_path is None:
local_dirname = "local_data"
if not os.path.exists(os.path.join(local_module, local_dirname)):
Expand Down
Loading
Loading