Contents:
models.nn.pyg_han
AttentionLayer
HANLayer
HAN
models.nn.pyg_han.
Bases: torch.nn.Module
torch.nn.Module
forward
Bases: models.BaseModel
models.BaseModel
add_args
Add model-specific arguments to the parser.
build_model_from_args
Build a new model instance.
loss
evaluate