antoniaebner commited on
Commit
c954357
·
verified ·
1 Parent(s): 8892856

Update src/train.py

Browse files
Files changed (1) hide show
  1. src/train.py +0 -1
src/train.py CHANGED
@@ -29,7 +29,6 @@ parser.add_argument(
29
  def main(args):
30
  print("Preprocess train molecules")
31
  data_path = os.path.join(DATA_FOLDER, "tox21_data.npz")
32
- data_path = "/system/user/studentwork/anebner/tox21_leaderboard/tox21_baselines/data_featset_new/tox21_descriptors.npz" # TMP override
33
  full_data = np.load(data_path, allow_pickle=True)
34
  features = full_data["features"]
35
  labels = full_data["labels"]
 
29
  def main(args):
30
  print("Preprocess train molecules")
31
  data_path = os.path.join(DATA_FOLDER, "tox21_data.npz")
 
32
  full_data = np.load(data_path, allow_pickle=True)
33
  features = full_data["features"]
34
  labels = full_data["labels"]