diff options
| author | Navan Chauhan <navanchauhan@gmail.com> | 2020-08-01 15:34:22 +0530 |
|---|---|---|
| committer | Navan Chauhan <navanchauhan@gmail.com> | 2020-08-01 15:34:22 +0530 |
| commit | e80badd278f54e8cde64243aa56687f8e6e32cc2 (patch) | |
| tree | 12667ca96236edca69b9f28ae7a66a1884f30156 /lstm_chem/trainer.py | |
| parent | cfa7ae936935010f77eaddd561ff1c0857a3c6f0 (diff) | |
added alpha model
Diffstat (limited to 'lstm_chem/trainer.py')
| -rw-r--r--[-rwxr-xr-x] | lstm_chem/trainer.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/lstm_chem/trainer.py b/lstm_chem/trainer.py index 4e8057e..0ed608a 100755..100644 --- a/lstm_chem/trainer.py +++ b/lstm_chem/trainer.py @@ -32,7 +32,8 @@ class LSTMChemTrainer(object): )) def train(self): - history = self.model.fit_generator( +# history = self.model.fit_generator( + history = self.model.fit( self.train_data_loader, steps_per_epoch=self.train_data_loader.__len__(), epochs=self.config.num_epochs, |