convnetjsとtensorflow その6

概要

tensorflowで学習したデータをconvnetjsで使ってみた。その2。
素数を聞いてみた。

写真

成果物

サンプルコード

var net;
function toarray(n) {
    var ary = new Array;
    for (var i = -1 ; i < 7; i++)
    {
        var s = 1;
        if (i > -1) s = 2 << i;
        var b = n & (s)
        if (b > 0)
        {
            ary.push(1.0);
        }
        else
        {
            ary.push(0.0);
        }
    }
    return ary;
}
function prime(j, w) {
    var str = "";
    if (w[1] > 0.9)
    {
        str = j + ", ";
    }
    else if (w[0] > 0.9)
    {
        str = "";
    }
    return str;
}
function test() {
    var str = "";
    var trainer = new convnetjs.Trainer(net);
    for (var j = 2; j < 101; j++) 
    {
        var point = new convnetjs.Vol(1, 1, 10);
        point.w = toarray(j);
        var prediction = net.forward(point);
        str += prime(j, prediction.w);
    }
    return str;
}
var load_from_json = function() {
    $.getJSON("/assets/q/o/E/T/qoETC", function(json) {
        net = new convnetjs.Net();
        net.fromJSON(json);
        var a = test();
        document.getElementById('helloWorld').innerHTML = a;
    });
}
load_from_json();

使用したライブラリ