Skip to content
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

improve-usability #81

Merged
merged 3 commits into from
Jan 19, 2025
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
small-fix
ryuta-yoshimatsu committed Jan 19, 2025
commit 264e55e0e7ea50a0117bf787934d59603690aeec
32 changes: 16 additions & 16 deletions examples/foundation_daily.py
Original file line number Diff line number Diff line change
@@ -44,7 +44,7 @@
# COMMAND ----------

# Number of time series
n = 1000
n = 100


def create_m4_daily():
@@ -115,21 +115,21 @@ def transform_group(df):
# COMMAND ----------

active_models = [
#"ChronosT5Tiny",
#"ChronosT5Mini",
#"ChronosT5Small",
#"ChronosT5Base",
#"ChronosT5Large",
#"ChronosBoltTiny",
#"ChronosBoltMini",
#"ChronosBoltSmall",
#"ChronosBoltBase",
#"MoiraiSmall",
#"MoiraiBase",
#"MoiraiLarge",
#"MoiraiMoESmall",
#"MoiraiMoEBase",
#"TimesFM_1_0_200m",
"ChronosT5Tiny",
"ChronosT5Mini",
"ChronosT5Small",
"ChronosT5Base",
"ChronosT5Large",
"ChronosBoltTiny",
"ChronosBoltMini",
"ChronosBoltSmall",
"ChronosBoltBase",
"MoiraiSmall",
"MoiraiBase",
"MoiraiLarge",
"MoiraiMoESmall",
"MoiraiMoEBase",
"TimesFM_1_0_200m",
"TimesFM_2_0_500m",
]

32 changes: 16 additions & 16 deletions examples/foundation_monthly.py
Original file line number Diff line number Diff line change
@@ -44,7 +44,7 @@
# COMMAND ----------

# Number of time series
n = 1000
n = 100


def create_m4_monthly():
@@ -126,21 +126,21 @@ def transform_group(df):
# COMMAND ----------

active_models = [
#"ChronosT5Tiny",
#"ChronosT5Mini",
#"ChronosT5Small",
#"ChronosT5Base",
#"ChronosT5Large",
#"ChronosBoltTiny",
#"ChronosBoltMini",
#"ChronosBoltSmall",
#"ChronosBoltBase",
#"MoiraiSmall",
#"MoiraiBase",
#"MoiraiLarge",
#"MoiraiMoESmall",
#"MoiraiMoEBase",
#"TimesFM_1_0_200m",
"ChronosT5Tiny",
"ChronosT5Mini",
"ChronosT5Small",
"ChronosT5Base",
"ChronosT5Large",
"ChronosBoltTiny",
"ChronosBoltMini",
"ChronosBoltSmall",
"ChronosBoltBase",
"MoiraiSmall",
"MoiraiBase",
"MoiraiLarge",
"MoiraiMoESmall",
"MoiraiMoEBase",
"TimesFM_1_0_200m",
"TimesFM_2_0_500m",
]

2 changes: 1 addition & 1 deletion examples/global_daily.py
Original file line number Diff line number Diff line change
@@ -45,7 +45,7 @@
# COMMAND ----------

# Number of time series
n = 1000
n = 100


def create_m4_daily():
2 changes: 1 addition & 1 deletion examples/global_monthly.py
Original file line number Diff line number Diff line change
@@ -44,7 +44,7 @@
# COMMAND ----------

# Number of time series
n = 1000
n = 100


def create_m4_monthly():
6 changes: 3 additions & 3 deletions examples/local_univariate_daily.py
Original file line number Diff line number Diff line change
@@ -51,7 +51,7 @@
# COMMAND ----------

# Number of time series
n = 1000
n = 100


def create_m4_daily():
@@ -151,8 +151,8 @@ def transform_group(df):
"RFableNNETAR",
"RFableEnsemble",
"RDynamicHarmonicRegression",
#"SKTimeTBats",
#"SKTimeLgbmDsDt",
"SKTimeTBats",
"SKTimeLgbmDsDt",
]

# COMMAND ----------
6 changes: 3 additions & 3 deletions examples/local_univariate_external_regressors_daily.py
Original file line number Diff line number Diff line change
@@ -58,7 +58,7 @@

# Number of time series to sample
sample = True
size = 1000
size = 100
stores = sorted(random.sample(range(0, 1000), size))

train = spark.read.csv(f"/Volumes/{catalog}/{db}/{volume}/train.csv", header=True, inferSchema=True)
@@ -126,8 +126,8 @@
"RFableNNETAR",
"RFableEnsemble",
"RDynamicHarmonicRegression",
#"SKTimeTBats",
#"SKTimeLgbmDsDt",
"SKTimeTBats",
"SKTimeLgbmDsDt",
]

# COMMAND ----------
6 changes: 3 additions & 3 deletions examples/local_univariate_monthly.py
Original file line number Diff line number Diff line change
@@ -52,7 +52,7 @@
# COMMAND ----------

# Number of time series
n = 1000
n = 100


def create_m4_monthly():
@@ -160,8 +160,8 @@ def transform_group(df):
"RFableNNETAR",
"RFableEnsemble",
"RDynamicHarmonicRegression",
#"SKTimeTBats",
#"SKTimeLgbmDsDt",
"SKTimeTBats",
"SKTimeLgbmDsDt",
]

# COMMAND ----------