Skip to content

Reset Labels for Labeled Nodes in LPA #2

@unbreakablegao

Description

@unbreakablegao

Hi,

In "model.py" when building lpa as following code, the labeled nodes seem not to be reset in the loop. I am wondering if this is correct. thank you!
def _build_lpa(self):
label_mask = tf.expand_dims(self.label_mask, -1)
input_labels = label_mask * self.labels
label_list = [input_labels]

    for _ in range(self.args.lpa_iter):
        lp_layer = LPALayer(adj=self.normalized_adj)
        hidden = lp_layer(label_list[-1])
        label_list.append(hidden)
    self.predicted_label = label_list[-1]

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions