Function prepUI

mnist-transfer-cnn/ui.js:27–35

(predict, retrain, testExamples, imageSize)

}

export function prepUI(predict, retrain, testExamples, imageSize) {
  setPredictFunction(predict, testExamples, imageSize);
  const imageInput = document.getElementById('image-input');
  imageInput.value = util.imageVectorToText(testExamples['5_1'], imageSize);
  predict(imageInput.value);
  setRetrainFunction(retrain);
  document.getElementById('retrain').disabled = false;
  document.getElementById('test-image-select').disabled = false;
}

export function getImageInput() {
  return document.getElementById('image-input').value;