从 tensorflow js 模型获取原始输出

Question

我正在通过一个简单的卷积网络连接到 MNIST 数据集。我想手动告诉程序完成该过程中的每个步骤，即遍历每个 Epoch 和批次，而不使用 model.fit（我想更好地了解 tensorflow 的内部工作原理）。

目前我有：

let EPOCHS = 10;    
let batches = Math.floor(TRAIN_DATA_SIZE/BATCH_SIZE);

      for (let i = 0; i < EPOCHS; i++){
        for (let j = 0; i < batches; i++){
          let inputs = await getNextTrain(BATCH_SIZE,TRAIN_DATA_SIZE, data);

          let inputXs = inputs[0];
          let inputYs = inputs[1];

          let output = await model.evaluate(inputXs,inputYs);

遍历每个 Epoch 和每个批次以进行 traian。但是 model.evalutate() 不返回网络的输出值，只返回定义的损失/指标。您是否可以为此添加特定指标以返回网络的输出

我想要的是代表网络认为每个输入的输出是什么的 10 元素数组（或张量）（使用 MNIST 所以希望概率网络认为每个输出是数字 0-9）

Answer 1

您可以在内部循环中调用 model.fit 并设置以下参数 epoch=1 initial_epoch=i \\ So that it trains only the current epoch and batch. 'i' being outer loop variable. x=inputXs y=inputYs

这将相应地更新模型中的所有权重。然后，您可以调用 model.evaluate 或 model.predict 或 model.get_layer 来获取您想要查看的信息。由于您现在拥有每个层的信息，您可以通过仅根据需要评估特定层来独立检查它们的输出值（参考您提到的行以查看所有概率......等等）

随着更新的变化：

let EPOCHS = 10;
let batches = Math.floor(TRAIN_DATA_SIZE/BATCH_SIZE);

  for (let i = 0; i < EPOCHS; i++){
    for (let j = 0; i < batches; i++){
      let inputs = await getNextTrain(BATCH_SIZE,TRAIN_DATA_SIZE, data);

      let inputXs = inputs[0];
      let inputYs = inputs[1];
      model.fit(..on above mentioned params..);

      model.evaluate() // To get loss and metric values.
      model.predict()  // To the final output of the model for input samples given.
      model.get_layer() //To get info about a particular layer and then retrieve the required info. Refer https://keras.io/layers/about-keras-layers/

      // For example to know the 1st layer's output : model.layers[0].output;

`