freedak/timesfm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge branch 'master' of https://github.com/google-research/timesfm into rajat_dev

Browse Source
This commit is contained in:
Rajat Sen
2025-01-15 21:24:14 +00:00
parent c49cf9caf8 a3ab7873e9
commit a9c5b48c49
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/timesfm/timesfm_base.py
+6 -6
View File
@@ -50,13 +50,13 @@ def moving_average(arr, window_size):
def freq_map(freq: str): def freq_map(freq: str):
"""Returns the frequency map for the given frequency string.""" """Returns the frequency map for the given frequency string."""
freq = str.upper(freq) freq = str.upper(freq)
if (freq.endswith("H") or freq.endswith("T") or freq.endswith("MIN") or if freq.endswith("MS"):
freq.endswith("D") or freq.endswith("B") or freq.endswith("U") or
freq.endswith("S")):
return 0
elif freq.endswith(("W", "M", "MS")):
return 1 return 1
elif freq.endswith("Y") or freq.endswith("Q") or freq.endswith("A"): elif freq.endswith(("H", "T", "MIN", "D", "B", "U", "S")):
return 0
elif freq.endswith(("W", "M")):
return 1
elif freq.endswith(("Y", "Q", "A")):
return 2 return 2
else: else:
raise ValueError(f"Invalid frequency: {freq}") raise ValueError(f"Invalid frequency: {freq}")