aboutsummaryrefslogtreecommitdiff
path: root/lstm_chem/model.py
diff options
context:
space:
mode:
Diffstat (limited to 'lstm_chem/model.py')
-rw-r--r--[-rwxr-xr-x]lstm_chem/model.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/lstm_chem/model.py b/lstm_chem/model.py
index 079589a..368a834 100755..100644
--- a/lstm_chem/model.py
+++ b/lstm_chem/model.py
@@ -4,7 +4,7 @@ from tensorflow.keras import Sequential
from tensorflow.keras.models import model_from_json
from tensorflow.keras.layers import LSTM, Dense
from tensorflow.keras.initializers import RandomNormal
-from lstm_chem.utils.smiles_tokenizer import SmilesTokenizer
+from lstm_chem.utils.smiles_tokenizer2 import SmilesTokenizer
class LSTMChem(object):