Spaces:
Running
Running
Update trainer.py
Browse files- trainer.py +1 -1
trainer.py
CHANGED
@@ -734,7 +734,7 @@ class Trainer(object):
|
|
734 |
|
735 |
|
736 |
if self.submodel == "NoTarget":
|
737 |
-
drug_smiles = [line for line in open("data/
|
738 |
else:
|
739 |
drug_smiles = [line for line in open("data/akt_train.smi", 'r').read().splitlines()]
|
740 |
|
|
|
734 |
|
735 |
|
736 |
if self.submodel == "NoTarget":
|
737 |
+
drug_smiles = [line for line in open("data/chembl_test.smi", 'r').read().splitlines()]
|
738 |
else:
|
739 |
drug_smiles = [line for line in open("data/akt_train.smi", 'r').read().splitlines()]
|
740 |
|