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の方も触ってみます。