File tree Expand file tree Collapse file tree 2 files changed +25
-0
lines changed Expand file tree Collapse file tree 2 files changed +25
-0
lines changed Original file line number Diff line number Diff line change
1
+ from class_resolver import Resolver
2
+
1
3
from .contextfeatureset import * # noqa:F401,F403
2
4
from .drugfeatureset import * # noqa:F401,F403
3
5
from .labeledtriples import * # noqa:F401,F403
4
6
from .datasetloader import * # noqa:F401,F403
5
7
from .drugpairbatch import * # noqa:F401,F403
6
8
from .batchgenerator import * # noqa:F401,F403
9
+
10
+ dataset_resolver = Resolver .from_subclasses (base = DatasetLoader )
Original file line number Diff line number Diff line change 6
6
from typing import Dict
7
7
from chemicalx .data import DrugFeatureSet , ContextFeatureSet , LabeledTriples
8
8
9
+ __all__ = [
10
+ "DatasetLoader" ,
11
+ "DrugCombDbDatasetLoader" ,
12
+ "DrugCombDatasetLoader" ,
13
+ ]
14
+
9
15
10
16
class DatasetLoader :
11
17
"""
@@ -102,3 +108,18 @@ def get_labeled_triples(self):
102
108
labeled_triples = LabeledTriples ()
103
109
labeled_triples .update_from_pandas (raw_data )
104
110
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" )
You can’t perform that action at this time.
0 commit comments