Files
pytorch-ts/pts/model/deepvar/deepvar_network.py
T
2020-01-02 12:44:04 +01:00

18 lines
352 B
Python

from typing import List, Optional, Tuple, Union
import torch
import torch.nn as nn
from torch.distributions import Distribution
import numpy as np
from pts.modules import DistributionOutput, MeanScaler, NOPScaler, FeatureEmbedder
class DeepVARTrainingNetwork(nn.Module):
pass
class DeepVARPredictionNetwork(DeepVARTrainingNetwork):
pass