wassname/pytorch-ts
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-ts.git synced 2026-06-27 18:06:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

norm_freq_str for fourier_time_features

Browse Source 
fixes issue #41
This commit is contained in:
Kashif Rasul
2021-02-21 11:49:56 +01:00
parent 9238fff0f5
commit 6eaac93f6f
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pts/feature/fourier_date_feature.py
+2 -2
View File
@@ -18,7 +18,7 @@ import pandas as pd
from pandas.tseries.frequencies import to_offset
from gluonts.core.component import validated
from gluonts.time_feature import TimeFeature
from gluonts.time_feature import TimeFeature, norm_freq_str
class FourierDateFeatures(TimeFeature):
@@ -50,7 +50,7 @@ class FourierDateFeatures(TimeFeature):
def fourier_time_features_from_frequency(freq_str: str) -> List[TimeFeature]:
offset = to_offset(freq_str)
multiple, granularity = offset.n, offset.name
granularity = norm_freq_str(offset.name)
features = {
"M": ["weekofyear"],