Skip to content

[NeurIPS '23] Evaluating Self-supervised Learning for Molecular Graph Embeddings

License

Notifications You must be signed in to change notification settings

hansen7/MolGraphEval

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

MolGraphEval

This repository is the official implementation of paper: "Evaluating Self-supervised Learning for Molecular Graph Embeddings”, NeurIPS 2023, Datasets and Benchmarks Track.

Diagram

Citation

@inproceedings{GraphEval,
  title = {Evaluating Self-supervised Learning for Molecular Graph Embeddings},
  author = {Hanchen Wang* and Jean Kaddour* and Shengchao Liu and Jian Tang and Joan Lasenby and Qi Liu},
  booktitle = {NeurIPS 2023, Datasets and Benchmarks Track},
  year = 2023
}

Usage

We include scripts for pre-training, probing and fine-tuning for GraphSSL on molecules, see script folder. We use conda to set up the environment:

conda env create -f env.yaml

About

[NeurIPS '23] Evaluating Self-supervised Learning for Molecular Graph Embeddings

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published