(exportPath)
| 152 | |
| 153 | // GRU with embedding. |
| 154 | async function exportGRUModel(exportPath) { |
| 155 | const model = tfl.sequential(); |
| 156 | const inputDim = 100; |
| 157 | model.add(tfl.layers.embedding({inputDim, outputDim: 20, inputShape: [10]})); |
| 158 | model.add(tfl.layers.gru({units: 4, goBackwards: true})); |
| 159 | |
| 160 | await saveModelAndRandomInputs(model, exportPath, inputDim); |
| 161 | } |
| 162 | |
| 163 | // Bidirecitonal LSTM with embedding. |
| 164 | async function exportBidirectionalLSTMModel(exportPath) { |
no test coverage detected
searching dependent graphs…