TensorFlow.jsをさわる

とりあえず触ってみました。 前提 インストール サンプルコードを動かす とりあえず何してるかは後で見るとして、動かしてみましょう。(コメントをGoogle翻訳してみました) 実行結果 Pythonの方も触ってみます。

広告ここから
広告ここまで

目次

    とりあえず触ってみました。

    前提

    $ node -v
    v9.11.1
    $ npm -v
    5.6.0

    インストール

    $ npm init -y
    $ npm install -S @tensorflow/tfjs

    サンプルコードを動かす

    とりあえず何してるかは後で見るとして、動かしてみましょう。(コメントをGoogle翻訳してみました)

    const tf = require('@tensorflow/tfjs');
    
    // Define a model for linear regression.
    // 線形回帰のモデルを定義します。
    const model = tf.sequential();
    model.add(tf.layers.dense({units: 1, inputShape: [1]}));
    
    // Prepare the model for training: Specify the loss and the optimizer.
    // トレーニングのためのモデルを準備する:損失とオプティマイザを指定します。
    model.compile({loss: 'meanSquaredError', optimizer: 'sgd'});
    
    // Generate some synthetic data for training.
    // トレーニング用の合成データを生成します。
    const xs = tf.tensor2d([1, 2, 3, 4], [4, 1]);
    const ys = tf.tensor2d([1, 3, 5, 7], [4, 1]);
    
    // Train the model using the data.
    // データを使用してモデルをトレーニングします。
    model.fit(xs, ys).then(() => {
       // Use the model to do inference on a data point the model hasn't seen before:
       // モデルが以前に見たことのないデータポイントの推論を行うためにモデルを使用する
       model.predict(tf.tensor2d([5], [1, 1])).print();
    });

    実行結果

    $ node index.js 
    Tensor
         [[8.3983145],]

    Pythonの方も触ってみます。

    広告ここから
    広告ここまで
    Home
    Search
    Bookmark