Function createDenseModel

tfjs-layers/src/engine/training_dataset_test.ts:25–30

()

import {FakeNumericDataset} from './dataset_fakes';

function createDenseModel(): tfl.LayersModel {
  const model = tfl.sequential();
  model.add(tfl.layers.dense(
      {units: 1, inputShape: [1], kernelInitializer: 'zeros'}));
  return model;
}

describeMathCPUAndGPU('LayersModel.fitDataset', () => {
  // Reference Python tf.keras code: