Sonja Topf
commited on
Commit
·
103d273
1
Parent(s):
a338012
remove import
Browse files- predict.py +2 -3
predict.py
CHANGED
|
@@ -1,7 +1,6 @@
|
|
| 1 |
import torch
|
| 2 |
import csv
|
| 3 |
import subprocess
|
| 4 |
-
import pandas as pd
|
| 5 |
|
| 6 |
from src.preprocess import create_clean_smiles
|
| 7 |
|
|
@@ -79,6 +78,6 @@ def predict(smiles_list):
|
|
| 79 |
|
| 80 |
return predictions
|
| 81 |
|
| 82 |
-
smiles = pd.read_csv("tox21_test.csv")["smiles"].tolist()
|
| 83 |
-
predict(smiles)
|
| 84 |
|
|
|
|
| 1 |
import torch
|
| 2 |
import csv
|
| 3 |
import subprocess
|
|
|
|
| 4 |
|
| 5 |
from src.preprocess import create_clean_smiles
|
| 6 |
|
|
|
|
| 78 |
|
| 79 |
return predictions
|
| 80 |
|
| 81 |
+
# smiles = pd.read_csv("tox21_test.csv")["smiles"].tolist()
|
| 82 |
+
# predict(smiles)
|
| 83 |
|