self.onmessage = function()

in web/src/webworker.js [4:24]

• 17 lines of code
• 2 McCabe index (conditional complexity)

self.onmessage = function(msg) {
    var classes = msg.data.classes;
    var data = msg.data.data;

    var tensorSize = data.length / (classes + 1);
    var size = Math.sqrt(tensorSize);
    var trainData = data.slice(0, tensorSize * classes);

    var parameters = trainedInit();
    var images = new jsnet.Tensor([classes, size, size, 1], trainData);
    var losses = trainNetwork(parameters, images, NUM_STEPS);

    images = new jsnet.Tensor([classes+1, size, size, 1], data);
    var output = applyNetwork(parameters, images).value;

    var probs = [];
    for (var i = 0; i < classes; ++i) {
        probs.push(Math.exp(output.data[i + classes * classes]));
    }
    self.postMessage({losses: losses, probs: probs});
};