plotting in scratch.py and trying TCN

2026-08-11 11:12:11 +08:00 · 2022-11-19 20:16:20 +08:00
parent 0a315b3d48
commit 90ee9c3298
11 changed files with 550 additions and 6 deletions
@@ -3,11 +3,14 @@ from typing import Union
 import torch

 from .DeepTIMe import deeptime
+from .DeepTIMe2 import deeptime2


 def get_model(model_type: str, **kwargs: Union[int, float]) -> torch.nn.Module:
    if model_type == 'deeptime':
        model = deeptime(datetime_feats=kwargs['datetime_feats'])
+    elif model_type=="deeptime2":
+        model = deeptime2(datetime_feats=kwargs['datetime_feats'])
    else:
        raise ValueError(f"Unknown model type {model_type}")
    return model