View source
on GitHub
The protocol message (protobuf) types defined by TensorFlow GNN.
This package is automatically included in the top-level tfgnn package:
import tensorflow_gnn as tfgnn
graph_schema = tfgnn.proto.GraphSchema()
Users are also free to import it separately as
import tensorflow_gnn.proto as tfgnn_proto
graph_schema = tfgnn_proto.GraphSchema()
...which, together with using its more targeted BUILD dependency, can help to shrink the bazel-bin/**/*.runfiles/ directory.
class BigQuery: Describes a BigQuery table or
SQL statement as datasource of a graph piece.
class Context: The schema for the features that
apply across the entire input graph.
class EdgeSet: The schema shared by a set of
edges that connect the same pair of node sets.
class Feature: The schema entry for a single
feature.
class GraphSchema: The top-level container
for the schema of a graph dataset.
class Metadata: Extra information optionally
provided on a context, node set or edge set.
class NodeSet: The schema shared by a set of
nodes in the graph.
class OriginInfo: Metadata about the origin of
the graph data.