huihun/graphsage-tf
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fixed error message when no features are present.

Browse Source
This commit is contained in:
williamleif 2017-09-16 14:26:47 -07:00
parent 87e978e415
commit b142770576
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
graphsage/models.py
View File

@ -231,7 +231,7 @@ class SampleAndAggregate(GeneralizedModel):
else: else:
self.embeds = None self.embeds = None
if features is None: if features is None:
if identity_dim is None: if identity_dim == 0:
raise Exception("Must have a positive value for identity feature dimension if no input features given.") raise Exception("Must have a positive value for identity feature dimension if no input features given.")
self.features = self.embeds self.features = self.embeds
else: else:

2
graphsage/supervised_models.py
View File

@ -53,7 +53,7 @@ class SupervisedGraphsage(models.SampleAndAggregate):
else: else:
self.embeds = None self.embeds = None
if features is None: if features is None:
if identity_dim is None: if identity_dim == 0:
raise Exception("Must have a positive value for identity feature dimension if no input features given.") raise Exception("Must have a positive value for identity feature dimension if no input features given.")
self.features = self.embeds self.features = self.embeds
else: else: