Function classify

intent-classifier/app/index.js:51–68

(sentences)

}

async function classify(sentences) {
  const [use, intent, metadata] = await Promise.all(
      [loadUSE(), loadIntentClassifer(DENSE_MODEL_URL), loadMetadata()]);

  const {labels} = metadata;
  const activations = await use.embed(sentences);

  const prediction = intent.predict(activations);

  const predsArr = await prediction.array();
  const preview = [predsArr[0].slice()];
  preview.unshift(labels);
  console.table(preview);

  tf.dispose([activations, prediction]);

  return predsArr[0];
}

const THRESHOLD = 0.90;
async function getClassificationMessage(softmaxArr) {