tensorflow.jsでsin その２

概要

tensorflow.jsでsin問題やってみた。
modelを使わないで、やってみた。

写真

サンプルコード

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 < 20; i++) 
    {
        ctx.lineTo(i * 10, hc - data[i] * 30);
    }
    ctx.stroke();
}
tx = [];
ty = [];
for (var i = 0; i < 20; i++) 
{
    var x = i / 3.0;    
    var y = Math.sin(x);
    tx.push(x);
    ty.push(y);
}
draw(ty, 0);
const train_x = tf.tensor1d(tx);
const train_y = tf.tensor1d(ty);
const w = tf.variable(tf.randomNormal([20], 0, 1));
const b = tf.variable(tf.zeros([20]));
const optimizer = tf.train.sgd(0.1);
function loss(pred, ypred) {
    const cost = pred.sub(ypred).square().mean();
    return cost;
}
function func(x) {
    return w.mul(x).add(b);
}
for (let i = 0; i < 200; i++) 
{
    optimizer.minimize(() => {
        const loss_var = loss(func(train_x), train_y);
        loss_var.print();
        return loss_var;
    })
}
const preds = func(train_x).dataSync();
draw(preds, 1);

成果物

以上。