1dbc4bf21b2f67189f3a0c761a774ccaadb47587,stellar/mapper/node_mappers.py,GraphSAGENodeMapper,__init__,#GraphSAGENodeMapper#Any#Any#Any#Any#Any#Any#Any#,49

Before Change

        // Ensure features are available:
        nodes_have_features = all(
            ["feature" in vdata for v, vdata in G.nodes(data=True)]
        )
        if not nodes_have_features:
            print("Warning: Not all nodes have a "feature" attribute.")

After Change

                )
            self.feature_size = feature_size

        elif len(feature_sizes) == 1:
            self.feature_size = int(feature_sizes.pop())

        else:
            raise RuntimeError(
                "Feature sizes in nodes inconsistent: "
                "found feature sizes: {}".format(feature_sizes)
            )

    def __len__(self):
        "Denotes the number of batches per epoch"
        return int(np.ceil(self.data_size / self.batch_size))

Instances