Function exportOneDimensionalModel

e2e/integration_tests/create_save_predict.js:189–200

(exportPath)

// Model with Conv1D and Pooling1D layers.
async function exportOneDimensionalModel(exportPath) {
  const model = tfl.sequential();
  model.add(tfl.layers.conv1d(
      {filters: 16, kernelSize: [4], inputShape: [80, 1], activation: 'relu'}));
  model.add(tfl.layers.maxPooling1d({poolSize: 3}));
  model.add(
      tfl.layers.conv1d({filters: 8, kernelSize: [3], activation: 'relu'}));
  model.add(tfl.layers.avgPooling1d({poolSize: 5}));
  model.add(tfl.layers.flatten());

  await saveModelAndRandomInputs(model, exportPath);
}

// Functional model with two Merge layers.
async function exportFunctionalMergeModel(exportPath) {