Function createDummyModel

tfjs-layers/src/callbacks_test.ts:17–22

()

describe('EarlyStopping', () => {
  function createDummyModel(): tfl.LayersModel {
    const model = tfl.sequential();
    model.add(tfl.layers.dense({units: 1, inputShape: [1]}));
    model.compile({loss: 'meanSquaredError', optimizer: 'sgd'});
    return model;
  }

  it('Default monitor, default mode, increasing val_loss', async () => {
    const model = createDummyModel();