Metadata-Version: 2.4
Name: dgf
Version: 0.0.3
Summary: Distributed Graph Flow
Home-page: https://github.com/google/dgf
Author: Mathieu Guillame-Bert, Brandon Mayer
License: Apache 2.0
Project-URL: Source, https://github.com/google/dgf.git
Project-URL: Tracker, https://github.com/google/dgf/issues
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: Science/Research
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Requires-Python: >=3.11
Description-Content-Type: text/markdown
License-File: LICENSE
Requires-Dist: numpy
Requires-Dist: absl-py
Requires-Dist: tensorflow
Requires-Dist: tensorflow-gnn
Requires-Dist: tf-keras
Requires-Dist: apache-beam[gcp]
Requires-Dist: graphviz
Requires-Dist: jax
Requires-Dist: flax
Requires-Dist: dataclasses-json
Requires-Dist: tqdm
Requires-Dist: bagz
Requires-Dist: dacite
Requires-Dist: google-cloud-spanner
Requires-Dist: google-api-core
Requires-Dist: google-cloud-storage
Requires-Dist: sparse-deferred
Requires-Dist: numba
Requires-Dist: tabulate
Requires-Dist: matplotlib
Requires-Dist: fsspec[gcs]
Requires-Dist: clu
Requires-Dist: orbax
Requires-Dist: frozendict
Requires-Dist: chex
Requires-Dist: etils
Requires-Dist: jaxtyping
Requires-Dist: altair
Requires-Dist: jinja2
Requires-Dist: reportlab
Requires-Dist: mmh3
Requires-Dist: google-api-core
Requires-Dist: pandas
Requires-Dist: PyYAML
Requires-Dist: tensorboard
Requires-Dist: immutabledict
Requires-Dist: networkx
Requires-Dist: pyarrow
Requires-Dist: fastavro
Requires-Dist: datasketch
Dynamic: author
Dynamic: classifier
Dynamic: description
Dynamic: description-content-type
Dynamic: home-page
Dynamic: license
Dynamic: license-file
Dynamic: project-url
Dynamic: requires-dist
Dynamic: requires-python
Dynamic: summary

# Distributed Graph Flow

<p align="center">
<img src="doc/docs/image/logo2-color-detoured-small.png" width="300" />
</p>

**(Distributed) Graph Flow** (GF) is a Python toolkit to develop and deploy
Graph Neural Network (**GNN**) models.

For more information, check the documentation at https://dgf.readthedocs.io/

This is not an officially supported Google product. This project is not
eligible for the [Google Open Source Software Vulnerability Rewards
Program](https://bughunters.google.com/open-source-security).

## Installation

To install YDF from [PyPI](https://pypi.org/project/dgf/), run:

```shell
pip install dgf -U
```

Currently, DGF is available on Python 3.11-13, on Linux x86-64.

## 😎 Minimal Usage example

```python
# Temporary fix for Keras dependency.
import os
os.environ["TF_USE_LEGACY_KERAS"] = "1"

# Import (distributed) graph flow
import dgf

# Fetch an example graph
graph, schema = dgf.io.fetch_ogb_graph("arxiv")

# Train a model
model = dgf.learning.train_node_model(graph=graph, schema=schema, target_column="labels")

# Look at the model
model.describe()

# Evaluate the model
model.evaluate()

# Make predictions
model.predict(graph, seed_node_idxs=[0, 1, 2])

# Save the model for later
model.save("/tmp/model")
```
