rezav
2 years ago
1 changed files with 1 additions and 1 deletions
+ 1
- 1
drug/datasets.py
View File
| class DDInteractionDataset(Dataset): | class DDInteractionDataset(Dataset): | ||||
| def __init__(self, root = "\\drug/data/", transform=None, pre_transform=None, pre_filter=None): | def __init__(self, root = "\\drug/data/", transform=None, pre_transform=None, pre_filter=None): | ||||
| super(DDInteractionDataset, self).__init__(os.path.dirname(os.path.abspath(os.path.dirname( __file__ ))) + "\\drug/data/", transform, pre_transform, pre_filter) | |||||
| super(DDInteractionDataset, self).__init__(os.path.dirname(os.path.abspath(os.path.dirname( __file__ ))) + "/drug/data/", transform, pre_transform, pre_filter) | |||||
| @property | @property | ||||
| def num_features(self): | def num_features(self): |
Loading…