Skip to content

Commit f4b5eff

Browse files
committed
update docs for datasets
1 parent 22d8371 commit f4b5eff

File tree

2 files changed

+23
-0
lines changed

2 files changed

+23
-0
lines changed

docs/source/dataset.rst

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,22 @@
1+
Datasets Overview
2+
====================================================
3+
4+
The ``torch-molecule`` provide easy access to a wide range of datasets (pre-cached or from Hugging Face Hub).
5+
6+
.. rubric:: QM9
7+
.. autofunction:: torch_molecule.datasets.load_qm9
8+
9+
.. rubric:: ChEMBL2k (a small subset from ChEMBL database)
10+
.. autofunction:: torch_molecule.datasets.load_chembl2k
11+
12+
.. rubric:: Broad6k
13+
.. autofunction:: torch_molecule.datasets.load_broad6k
14+
15+
.. rubric:: ToxCast (The US EPA Toxicity Forecaster program)
16+
.. autofunction:: torch_molecule.datasets.load_toxcast
17+
18+
.. rubric:: ADMET (absorption-distribution-metabolism-excretion-toxicity)
19+
.. autofunction:: torch_molecule.datasets.load_admet
20+
21+
.. rubric:: Gas Permeability for polymers
22+
.. autofunction:: torch_molecule.datasets.load_gasperm

docs/source/index.rst

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@ It offers a user-friendly interface similar to `sklearn`, and includes model che
1919

2020
install
2121
example
22+
dataset
2223
overview
2324

2425
.. toctree::

0 commit comments

Comments
 (0)