Skip to content

Commit

Permalink
Fixed sampler guide example for crediccard using DirectRunner
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 590612793
  • Loading branch information
aferludin authored and tensorflower-gardener committed Dec 13, 2023
1 parent 2e399e2 commit bf6c785
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions tensorflow_gnn/docs/guide/sampler.md
Original file line number Diff line number Diff line change
Expand Up @@ -128,6 +128,7 @@ Beam direct runner.
```
cd <path-to>/gnn/examples/sampler/creditcard
python3 -m tensorflow_gnn.experimental.sampler.beam.sampler \
--data_path="." \
--graph_schema graph_schema.pbtxt \
--sampling_spec sampling_spec.pbtxt \
--output_samples outputs/examples.tfrecord \
Expand Down

0 comments on commit bf6c785

Please sign in to comment.