Join us at TensorFlow World, Oct 28-31. Use code TF20 for 20% off select passes. Register now


View source on GitHub

Tools and APIs for preparing data for Neural Structured Learning.


build_graph module: Program to build a graph based on dense input features (embeddings).

graph_utils module: Utility functions for manipulating (weighted) graphs.

pack_nbrs module: Tool that prepares input for graph-based Neural Structured Learning.


add_edge(...): Adds an edge to a given graph.

add_undirected_edges(...): Makes all edges of the given graph bi-directional.

read_tsv_graph(...): Reads the file filename containing graph edges in TSV format.

write_tsv_graph(...): Writes the given graph to the file filename in TSV format.