Skip to content

drorlab/atom3d

Folders and files

NameName
Last commit message
Last commit date

Latest commit

4c2f3b7 · Mar 2, 2023
Feb 24, 2023
Mar 2, 2023
Jan 15, 2022
Jul 5, 2022
Jun 4, 2021
Jul 3, 2021
May 11, 2020
Oct 2, 2020
Aug 12, 2021
Jan 16, 2021
Aug 8, 2021
Jul 23, 2020
Aug 18, 2021
Jun 27, 2021
Dec 23, 2020
Jan 16, 2021
Jun 27, 2021
Jul 19, 2022

Repository files navigation

ATOM3D: Tasks On Molecules in 3 Dimensions

Documentation Status Package codecov PyPI version

ATOM3D enables machine learning on three-dimensional molecular structure.

Features

  • Access to several datasets involving 3D molecular structure.
  • LMDB data format for storing lots of molecules (and associated metadata).
  • Utilities for splitting/filtering data based on many criteria.

For more detailed information, read the documentation.

Installation

Install with:

pip install atom3d

To use rdkit functionality, please install within conda:

conda create -n atom3d python=3.6 pip rdkit
conda activate atom3d
pip install atom3d

Usage

Downloading a dataset

From python:

import atom3d.datasets as da
da.download_dataset('lba', PATH_TO_DATASET) # Download LBA dataset.

Or, download and unzip from the website.

Loading a dataset

From python:

import atom3d.datasets as da
dataset = da.load_dataset(PATH_TO_DATASET, {'lmdb','pdb','silent','sdf','xyz','xyz-gdb'})
print(len(dataset))  # Print length
print(dataset[0].keys())  # Print keys

LMDB datasets

LMDB allows for compressed, fast, random access to your structures, all within a single database. Currently, we support creating LMDB datasets from PDB files, silent files, and xyz files.

Creating an LMDB dataset

From command line:

python -m atom3d.datasets PATH_TO_PDB_DIR PATH_TO_DATASET --filetype {pdb,silent,xyz,xyz-gdb} 

For more usage, please see the documentation.

Contribute

As a living repository, we welcome contributions of additional datasets, methods, and functionality! See the Contributing section of the documentation for details.

Support

For support, please file an issue at https://github.com/drorlab/atom3d/issues.

License

The project is licensed under the MIT license.

Reference

We provide an overview on ATOM3D and details on the preparation of all datasets in our preprint:

R. J. L. Townshend, M. Vögele, P. Suriana, A. Derry, A. Powers, Y. Laloudakis, S. Balachandar, B. Jing, B. Anderson, S. Eismann, R. Kondor, R. B. Altman, R. O. Dror "ATOM3D: Tasks On Molecules in Three Dimensions", arXiv:2012.04035

Please cite this work if some of the ATOM3D code or datasets are helpful in your scientific endeavours. For specific datasets, please also cite the respective original source(s), given in the preprint.