ethanfetaya / NRI

Neural relational inference for interacting systems - pytorch
MIT License
739 stars 157 forks source link

non-interaction edge type #33

Open fatcatZF opened 3 years ago

fatcatZF commented 3 years ago

For the system in which 2 particles interact or not, such as the spring experiments, if we use z{ij}=[0,1] to denote interaction, and z{ij} = [1,0] to denote non-interaction(no message between node i and j), in the decoder should we only consider the interaction edge type, i.e., h^t(i,j) = z{ij,0}fe([x^t_i, x^t_j])? Since no message between the non-interaction edge.