Open yue0708 opened 5 months ago
lass GCN(nn.Module): def __init__(self, adj=np.load('/home/zcy/latent-diffusion-main/datasets/traffic/matrix.npy'), input_dim = 3, hidden_dim =3, output_dim =3): super(GCN, self).__init__() self.register_buffer( "laplacian", calculate_laplacian_with_self_loop(th.FloatTensor(adj).to(device=device)) ) self._num_nodes = adj.shape[0] self._input_dim = input_dim # seq_len for prediction self._hidden_dim = hidden_dim self._output_dim = output_dim # output_dim for prediction self.weights1 = nn.Parameter( th.FloatTensor(self._input_dim, self._hidden_dim) ) 问下这里的matrix文件有提供吗：/home/zcy/latent-diffusion-main/datasets/traffic/matrix.npy ？我在数据集和代码都没找到
lass GCN(nn.Module): def __init__(self, adj=np.load('/home/zcy/latent-diffusion-main/datasets/traffic/matrix.npy'), input_dim = 3, hidden_dim =3, output_dim =3): super(GCN, self).__init__() self.register_buffer( "laplacian", calculate_laplacian_with_self_loop(th.FloatTensor(adj).to(device=device)) ) self._num_nodes = adj.shape[0] self._input_dim = input_dim # seq_len for prediction self._hidden_dim = hidden_dim self._output_dim = output_dim # output_dim for prediction self.weights1 = nn.Parameter( th.FloatTensor(self._input_dim, self._hidden_dim) )
再找一下通讯作者重新发给你，可能遗漏了
ChyaZhang
commented
2 months ago ChyaZhang
commented
2 months ago
lass GCN(nn.Module): def __init__(self, adj=np.load('/home/zcy/latent-diffusion-main/datasets/traffic/matrix.npy'), input_dim = 3, hidden_dim =3, output_dim =3): super(GCN, self).__init__() self.register_buffer( "laplacian", calculate_laplacian_with_self_loop(th.FloatTensor(adj).to(device=device)) ) self._num_nodes = adj.shape[0] self._input_dim = input_dim # seq_len for prediction self._hidden_dim = hidden_dim self._output_dim = output_dim # output_dim for prediction self.weights1 = nn.Parameter( th.FloatTensor(self._input_dim, self._hidden_dim) )问下这里的matrix文件有提供吗：/home/zcy/latent-diffusion-main/datasets/traffic/matrix.npy ？我在数据集和代码都没找到