Belle II Software development
GraFEIModel Member List

This is the complete list of members for GraFEIModel, including all inherited members.

__init__(self, nfeat_in_dim, efeat_in_dim, gfeat_in_dim, edge_classes=6, x_classes=7, hidden_layer_dim=128, num_hid_layers=1, num_ML=1, dropout=0.0, global_layer=True, **kwargs)GraFEIModel
first_MLGraFEIModel
forward(self, batch)GraFEIModel
last_MLGraFEIModel
ML_listGraFEIModel