metric-few-shot-graph

KO | EN

gitlite — search

#javascript #python #hacktoberfest #android #deep-learning #linux #java #awesome #kubernetes #vue #pytorch #openai #ios #open-source

metric-few-shot-graph

(★ 24)

Few-Shot Graph Classification via distance metric learning.

metric-few-shot-graph Latest Version Download

Download Latest Version (.zip)

// 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

metric-few-shot-graph Latest Version Download

Today's picks

Was this content helpful?

Recent Feedback

metric-few-shot-graph

metric-few-shot-graph Latest Version Download

Today's picks

Was this content helpful?

Recent Feedback

Download README