wassname/DeepTime
1
0
Fork 0
mirror of https://github.com/wassname/DeepTime.git synced 2026-07-28 11:14:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

single variate

Browse Source
This commit is contained in:
wassname
2022-11-20 14:29:25 +08:00
parent d4e0cda175
commit efbf767523
4 changed files with 559 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models/DeepTIMe2.py
+6 -3
View File
@@ -21,10 +21,13 @@ def deeptime2(datetime_feats: int, layer_size: int, inr_layers: int, n_fourier_f
class DeepTIMe2(nn.Module):
def __init__(self, datetime_feats: int, layer_size: int, inr_layers: int, n_fourier_feats: int, scales: float):
def __init__(self, datetime_feats: int, layer_size: int, inr_layers: int, n_fourier_feats: int, scales: float, dropout: float=0.3):
super().__init__()
self.inr = INRPlus2(in_feats=datetime_feats + 1, layers=inr_layers, layer_size=layer_size,
n_fourier_feats=n_fourier_feats, scales=scales)
in_feats=datetime_feats
if n_fourier_feats:
in_feats += 1
self.inr = INRPlus2(in_feats=in_feats, layers=inr_layers, layer_size=layer_size,
n_fourier_feats=n_fourier_feats, scales=scales, dropout=dropout)
self.adaptive_weights = RidgeRegressor()
self.datetime_feats = datetime_feats