plunkerでtensorflow.js その17

概要

plunkerでtensorflow.jsやってみた。
lstm使ってみた。
sin波、学習させてみた。

サンプルコード

tf.setBackend('cpu');
var canvas = document.getElementById("canvas");
var ctx = canvas.getContext("2d");
function draw(data, n) {
    var hc = n * 100 + 100;
    ctx.strokeStyle = "#f00";
    ctx.lineWidth = 1;
    ctx.moveTo(0, hc);
    for (var i = 1; i < 200; i++)
    {
        ctx.lineTo(i, hc - data[i] * 30);
    }
    ctx.stroke();
}

const model = tf.sequential();
model.add(tf.layers.lstm({ 
  units: 64,
  returnSequences: false,
  inputShape: [2, 1]
}));
model.add(tf.layers.dense({
  units: 1, 
  activation: "tanh"
}));
model.compile({
  optimizer: 'adam',
  loss: 'meanSquaredError'
});

const buf0 = [];
const buf1 = [];
b = [];
for (var i = 0; i < 200; i++)
{
    var x = i / 30.0;
    var y = Math.sin(x);
    b.push(y);
}
for (var i = 0; i < 197; i++)
{
    buf0.push([[b[i]], [b[i + 1]]]);
    buf1.push([b[i + 2]]);
}
draw(b, 0);

const xs = tf.tensor3d(buf0);
const ys = tf.tensor2d(buf1);
model.fit(xs, ys, {
    batchSize: 100,
    epochs: 100
}).then((d) => {
    var str = "loss = ";
    str += d.history.loss[0];
    alert(str);
    p = [];
    for (i = 0; i < 197; i++)
    {
        var x =  i / 30.0;
        var y0 = Math.sin(x);
        x =  (i + 1) / 30.0;
        var y1 = Math.sin(x);
        var pre = model.predict(tf.tensor3d([[[y0], [y1]]], [1, 2, 1]));
        var f = pre.dataSync();
        p.push(f);
    }
    draw(p, 1);
});

成果物

比較用

tanh
https://embed.plnkr.co/oVEtQXFPuAz2hLONXL4g/

以上。