updating benchmarks to point to v2.0

This commit is contained in:
Rajat Sen
2025-01-08 18:51:47 +00:00
parent 1e249ef0b1
commit 8fe4a0a511
3 changed files with 38 additions and 14 deletions
+3
View File
@@ -636,6 +636,7 @@ class TimesFmBase:
model_name: str = "timesfm",
window_size: int | None = None,
num_jobs: int = 1,
normalize: bool = False,
verbose: bool = True,
) -> pd.DataFrame:
"""Forecasts on a list of time series.
@@ -654,6 +655,7 @@ class TimesFmBase:
window_size: window size of trend + residual decomposition. If None then
we do not do decomposition.
num_jobs: number of parallel processes to use for dataframe processing.
normalize: normalize context before forecasting or not.
verbose: output model states in terminal.
Returns:
@@ -698,6 +700,7 @@ class TimesFmBase:
freq_inps = [freq_map(freq)] * len(new_inputs)
_, full_forecast = self.forecast(new_inputs,
freq=freq_inps,
normalize=normalize,
window_size=window_size)
if verbose:
print("Finished forecasting.")