Skip to content

Commit e3e9030

Browse files
committed
Update GraphSAGE docs in README to tuple return
1 parent e45c44c commit e3e9030

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

README.md

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -184,8 +184,9 @@ Please see the GDS documentation for more on the pipelines' procedure APIs.
184184
Assuming we have a graph `G` with node property `x`, we can do the following:
185185

186186
```python
187-
model = gds.beta.graphSage.train(G, modelName="myModel", featureProperties=["x"])
187+
model, res = gds.beta.graphSage.train(G, modelName="myModel", featureProperties=["x"])
188188
assert len(model.metrics()["epochLosses"]) == model.metrics()["ranEpochs"]
189+
assert res["trainMillis"] >= 0
189190

190191
res = model.predict_stream(G)
191192
assert len(res) == G.node_count()

0 commit comments

Comments
 (0)