Function exportSimpleRNNModel

e2e/integration_tests/create_save_predict.js:144–151

(exportPath)

// SimpleRNN with embedding.
async function exportSimpleRNNModel(exportPath) {
  const model = tfl.sequential();
  const inputDim = 100;
  model.add(tfl.layers.embedding({inputDim, outputDim: 20, inputShape: [10]}));
  model.add(tfl.layers.simpleRNN({units: 4}));

  await saveModelAndRandomInputs(model, exportPath, inputDim);
}

// GRU with embedding.
async function exportGRUModel(exportPath) {