
630 Глава 18
ектах данных для их правильной обработки. В частности,
x
должен содержать признаки
узла,
edge_attr
— признаки ребер,
edge_index
— список ребер, а
y
— метки. Данные QM9
содержат некоторые дополнительные примечательные атрибуты:
pos
— положение ка-
ждого атома молекулы в трехмерной сетке и
z
— атомный номер каждого атома в мо-
лекуле. Метки в QM9 представляют собой набор физических свойств молекул — таких
как дипольный момент, свободная энергия, энтальпия или изотропная поляризация. Мы
собираемся построить GNN и обучить ее на QM9 для предсказания изотропной поляри-
зации.
Набор данных QM9
Набор данных QM9 содержит описания 133 885 ...