Function indexToOneHot

mnist-transfer-cnn/util.js:75–81

(index, numClasses)

}

export function indexToOneHot(index, numClasses) {
  const oneHot = [];
  for (let i = 0; i < numClasses; ++i) {
    oneHot.push(i === index ? 1 : 0);
  }
  return oneHot;
}

export function convertDataToTensors(data, numClasses) {
  const numExamples = data.length;