diff options
| author | Navan Chauhan <navanchauhan@gmail.com> | 2020-09-24 16:11:25 +0530 |
|---|---|---|
| committer | Navan Chauhan <navanchauhan@gmail.com> | 2020-09-24 16:11:25 +0530 |
| commit | 7e9940d6c7fcca0109e90b3d68ecd68676635119 (patch) | |
| tree | 06d1ba55dd5ac24f86021ab59994114e6c4ac7e9 /app/prod/model_arch.json | |
| parent | ebd31b9f60ea93391e3eb7d58c1a6fe0b813c6a9 (diff) | |
replaced dataset with one trained on 1.5 mil compounds
Diffstat (limited to 'app/prod/model_arch.json')
| -rw-r--r-- | app/prod/model_arch.json | 21 |
1 files changed, 15 insertions, 6 deletions
diff --git a/app/prod/model_arch.json b/app/prod/model_arch.json index 561b34f..aaaf55e 100644 --- a/app/prod/model_arch.json +++ b/app/prod/model_arch.json @@ -4,6 +4,20 @@ "name": "sequential", "layers": [ { + "class_name": "InputLayer", + "config": { + "batch_input_shape": [ + null, + null, + 47 + ], + "dtype": "float32", + "sparse": false, + "ragged": false, + "name": "lstm_input" + } + }, + { "class_name": "LSTM", "config": { "name": "lstm", @@ -137,13 +151,8 @@ "bias_constraint": null } } - ], - "build_input_shape": [ - null, - null, - 47 ] }, - "keras_version": "2.3.0-tf", + "keras_version": "2.4.0", "backend": "tensorflow" }
\ No newline at end of file |