打印所有图层输出

时间:2018-07-23 16:19:33

标签: javascript tensorflow.js

给出以下模型,如何打印所有图层值?

const input = tf.input({shape: [5]});
    const denseLayer1 = tf.layers.dense({units: 10, activation: 'relu'});
    const denseLayer2 = tf.layers.dense({units: 2, activation: 'softmax'});
    const output = denseLayer2.apply(denseLayer1.apply(input));
    const model = tf.model({inputs: input, outputs: output});
    model.predict(tf.ones([2, 5])).print();
    
    
 
<html>
  <head>
    <!-- Load TensorFlow.js -->
    <script src="https://cdn.jsdelivr.net/npm/@tensorflow/tfjs@0.12.0"> </script>
  </head>

  <body>
  </body>
</html>

2 个答案:

答案 0 :(得分:0)

您可以这样

    for(var i = 0; i < tf.layers.length; i++)
      model.predict(tf.layers[i].value).print();
      // OR
      model.predict(tf.layers[i].inputs).print();

我不知道您的数组的结构如何,但是类似的方法可能有用。

答案 1 :(得分:0)

要打印图层,需要使用<link rel="stylesheet" href="https://stackpath.bootstrapcdn.com/bootstrap/4.1.1/css/bootstrap.min.css"> <script src="https://code.jquery.com/jquery-3.3.1.min.js"></script> <script src="https://stackpath.bootstrapcdn.com/bootstrap/4.1.1/js/bootstrap.bundle.min.js"></script> <form name="formFiltre" id="formFiltre" method="post"> <div class="form-inline"> <div class="form-group "> <input type="hidden" name="nav" value="1" /> <label for="rf_numero" class="col-form-label sr-only">Numéro</label> <input type="text" name="rf_numero" id="rf_numero" class="form-control mb-2 mr-sm-2" placeholder="Numéro" value="" /> </div> <div class="form-group"> <label for="rf_client" class="form-label sr-only">Client</label> <input type="text" name="rf_client" id="rf_client" class="form-control mb-2 mr-sm-2" placeholder="Client" value="" /> </div> </div> <div class="form-inline"> <div class="form-group"> <label for="statuts" class="col-form-label sr-only">Statut</label> <select name="rf_statut_commande" id="rf_statut_commande" class="form-control mb-2 mr-sm-2"> <option value="">-- Commande --</option> <option value="En attente de paiement">En attente de paiement</option> <option value="Paiement effectué">Paiement effectué</option> <option value="Traitement en cours">Traitement en cours</option> <option value="Expedié">Expediée</option> <option value="Terminée">Terminée</option> <option value="Annulée">Annulée</option> </select> </div> <div class="form-group"> <select name="rf_reglement_statut" id="rf_reglement_statut" class="form-control mb-2 mr-sm-2"> <option value="">-- Règlement --</option> <option value="En attente">En attente</option> <option value="Effectué">Effectué</option> <option value="Annulé">Annulé</option> </select> <div class="mb-2 mr-sm-2"> <input type="submit" class="btn" value="Rechercher" /> </div> </div> </div> </form>属性定义要在模型配置中输出的图层。在outputs上使用解构分配,可以检索中间层以进行输出

model.predict()
const input = tf.input({shape: [5]});
        const denseLayer1 = tf.layers.dense({units: 10, activation: 'relu'});
        const denseLayer2 = tf.layers.dense({units: 2, activation: 'softmax'});
        const output1 = denseLayer1.apply(input);
        const output2 = denseLayer2.apply(output1);
        const model = tf.model({inputs: input, outputs: [output1, output2]});
        const [firstLayer, secondLayer] = model.predict(tf.ones([2, 5]));
        firstLayer.print();
        secondLayer.print()