diff options
Diffstat (limited to 'lstm_chem/finetuner.py')
| -rw-r--r--[-rwxr-xr-x] | lstm_chem/finetuner.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/lstm_chem/finetuner.py b/lstm_chem/finetuner.py index 904958b..24f26ce 100755..100644 --- a/lstm_chem/finetuner.py +++ b/lstm_chem/finetuner.py @@ -1,4 +1,4 @@ -from lstm_chem.utils.smiles_tokenizer import SmilesTokenizer +from lstm_chem.utils.smiles_tokenizer2 import SmilesTokenizer from lstm_chem.generator import LSTMChemGenerator @@ -14,7 +14,8 @@ class LSTMChemFinetuner(LSTMChemGenerator): self.model.compile(optimizer=self.config.optimizer, loss='categorical_crossentropy') - history = self.model.fit_generator( +# history = self.model.fit_generator( + history = self.model.fit( self.finetune_data_loader, steps_per_epoch=self.finetune_data_loader.__len__(), epochs=self.config.finetune_epochs, |