Browse Source

path small change

main
rezav 2 years ago
parent
commit
8b3ef6ccb6
1 changed files with 3 additions and 2 deletions
  1. 3
    2
      drug/datasets.py

+ 3
- 2
drug/datasets.py View File

@@ -4,12 +4,13 @@ import torch
from torch_geometric.data import Dataset, Data
import numpy as np
import random
import os



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
def num_features(self):

Loading…
Cancel
Save