rezav
2 years ago
1 changed files with 3 additions and 2 deletions
+ 3
- 2
drug/datasets.py
View File
| from torch_geometric.data import Dataset, Data | from torch_geometric.data import Dataset, Data | ||||
| import numpy as np | import numpy as np | ||||
| import random | import random | ||||
| import os | |||||
| class DDInteractionDataset(Dataset): | class DDInteractionDataset(Dataset): | ||||
| def __init__(self, root = "drug/data/", transform=None, pre_transform=None, pre_filter=None): | |||||
| super(DDInteractionDataset, self).__init__(root, transform, pre_transform, pre_filter) | |||||
| 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) | |||||
| @property | @property | ||||
| def num_features(self): | def num_features(self): |
Loading…