def __init__(self, opt, kernel_size=2, n_blocks=1, state_dim_bottleneck=64, annotation_dim_bottleneck=64):
super(STGCN, self).__init__()
self.batchSize = opt.batchSize
self.state_dim = opt.state_dim
self.n_node = opt.n_node
# ST-Block
self.st_blocks = ST_blocks(opt, kernel_size=kernel_size, n_blocks=n_blocks, state_dim_bottleneck=state_dim_bottleneck, annotation_dim_bottleneck=annotation_dim_bottleneck)
opt.L = opt.L - 2 * n_blocks * (kernel_size - 1)
# GCNN ()
self.gcnn = GatedCNN(opt, in_channels=self.state_dim, out_channels=self.state_dim, kernel_size=opt.L)
opt.L = 1
# FC ()
self.out = nn.Sequential(
nn.Linear(self.state_dim, opt.output_dim),
nn.Sigmoid()
)
opt.L = opt.init_L
self._initialization()