Skip to content

Commit ddb47ea

Browse files
Merge pull request #24 from cthoyt/resolve-datasets
Add dataset resolver around dataset loader class
2 parents 732660a + 2987385 commit ddb47ea

File tree

2 files changed

+25
-0
lines changed

2 files changed

+25
-0
lines changed

chemicalx/data/__init__.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,10 @@
1+
from class_resolver import Resolver
2+
13
from .contextfeatureset import * # noqa:F401,F403
24
from .drugfeatureset import * # noqa:F401,F403
35
from .labeledtriples import * # noqa:F401,F403
46
from .datasetloader import * # noqa:F401,F403
57
from .drugpairbatch import * # noqa:F401,F403
68
from .batchgenerator import * # noqa:F401,F403
9+
10+
dataset_resolver = Resolver.from_subclasses(base=DatasetLoader)

chemicalx/data/datasetloader.py

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,12 @@
66
from typing import Dict
77
from chemicalx.data import DrugFeatureSet, ContextFeatureSet, LabeledTriples
88

9+
__all__ = [
10+
"DatasetLoader",
11+
"DrugCombDbDatasetLoader",
12+
"DrugCombDatasetLoader",
13+
]
14+
915

1016
class DatasetLoader:
1117
"""
@@ -102,3 +108,18 @@ def get_labeled_triples(self):
102108
labeled_triples = LabeledTriples()
103109
labeled_triples.update_from_pandas(raw_data)
104110
return labeled_triples
111+
112+
113+
class DrugCombDbDatasetLoader(DatasetLoader):
114+
"""A dataset loader for `DrugCombDB <http://drugcombdb.denglab.org>`_."""
115+
116+
def __init__(self):
117+
super().__init__("drugcombdb")
118+
119+
120+
# FIXME need reference
121+
class DrugCombDatasetLoader(DatasetLoader):
122+
"""A dataset loader for DrugComb."""
123+
124+
def __init__(self):
125+
super().__init__("drugcomb")

0 commit comments

Comments
 (0)