Сопоставление тензорного потока js вероятностных предсказаний с их метками - PullRequest
0 голосов
/ 26 апреля 2020

Я могу предсказывать и получать вероятности, но я хочу иметь возможность печатать фактические метки, связанные с вероятностями вероятности. Не могли бы вы помочь

      const image = tf.browser.fromPixels(imgcanvas);

      const resized_image = 

           tf.image.resizeBilinear(image, [128,128]).toFloat();

      const offset = tf.scalar(255.0);

      const normalized = tf.scalar(1.0).sub(resized_image.div(offset));

      const batchedImage = normalized.expandDims(0);

      const MODEL_URL = 'tfjs_model/model.json';

      const model = await tf.loadLayersModel(MODEL_URL);

      const result = model.predict(batchedImage);

      result.print();
...