Function createModelForTest

tfjs-node/src/tensorboard_test.ts:230–238

()

  });

  function createModelForTest(): tfn.LayersModel {
    const model = tfn.sequential();
    model.add(
        tfn.layers.dense({units: 5, activation: 'relu', inputShape: [10]}));
    model.add(tfn.layers.dense({units: 1}));
    model.compile(
        {loss: 'meanSquaredError', optimizer: 'sgd', metrics: ['MAE']});
    return model;
  }

  it('fit(): default epoch updateFreq, with validation', async () => {
    const model = createModelForTest();