The atom prediction module takes the hidden state of the graph as input, and outputs the type of atom as a probability by passing it through the fully connected layer (F C a (•)). The dimension of the output is (atom type) +1, which is the sum of the number of atom types and the label indicating the termination.