Shortcuts

Graph Embedder

The Task

This task consists of creating an embedding of a graph. That is, a vector of features which can be used for a downstream task. The GraphEmbedder and GraphClassificationData classes internally rely on pytorch-geometric.


Example

Let’s look at generating embeddings of graphs from the KKI data set from TU Dortmund University.

We start by creating the TUDataset <https://pytorch-geometric.readthedocs.io/en/latest/_modules/torch_geometric/datasets/tu_dataset.html#TUDataset>. Next, we load a trained GraphEmbedder (from a previously trained GraphClassifier). Finally, we save the model. Here’s the full example:

import flash

To learn how to view the available backbones / heads for this task, see Backbones and Heads.

Read the Docs v: latest
Versions
latest
stable
0.8.2
0.8.1.post0
0.8.1
0.8.0
0.7.5
0.7.4
0.7.3
0.7.2
0.7.1
0.7.0
0.6.0
0.5.2
0.5.1
0.5.0
0.4.0
0.3.2
0.3.1
0.3.0
0.2.3
0.2.2
0.2.1
0.2.0
0.1.0post1
Downloads
html
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.