metric-few-shot-graph

KO | EN

gitlite — search

#python #android #java #tensorflow #javascript #machine-learning #react #kafka #pytorch #iot #rest-api #linux

metric-few-shot-graph

★ 24 Open GitHub ↗

Few-Shot Graph Classification via distance metric learning.

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# Metric Based Few-Shot Graph Classification

Codebase for the paper [Metric Based Few-Shot Graph Classification](https://proceedings.mlr.press/v198/crisostomi22a.html), published at Learning on Graphs (2022).

## Installation

Setup the development environment:

```bash
conda create --name fs-grl python=3.9
conda activate fs-grl
```
Install PyTorch with CUDA support according to https://pytorch.org/get-started/locally/.

Install PyG
```bash
conda install pyg -c pyg
```

Install the project in edit mode:
```bash
pip install -e .
```

### Download data

Download the versioned datasets:
```bash
dvc pull
dvc checkout
```

### Training a model
You can train and evaluate various families of models by running the corresponding script in the `scripts` folder. For example, to train a Distance Metric Learning model, you can run:
```bash
python fs_grl/scripts/run_dml.py
```

metric-few-shot-graph

Explore Similar Repositories

Was this content helpful?

Recent Feedback

Download README