diff --git a/graphsage/models.py b/graphsage/models.py index b3b9db45..6b55c47c 100644 --- a/graphsage/models.py +++ b/graphsage/models.py @@ -219,7 +219,7 @@ def __init__(self, placeholders, features, adj, degrees, elif aggregator_type == "gcn": self.aggregator_cls = GCNAggregator else: - raise Exception("Unknown aggregator: ", self.aggregator_cls) + raise Exception("Unknown aggregator: ", aggregator_type) # get info from placeholders... self.inputs1 = placeholders["batch1"]