def get_atom_features(self, node_id):
attrs = nx.get_node_attributes(self.graph, "atom_features")
return attrs[node_id]
微信
公众号
扫码关注公众号
微信
公众号