Tensorflow.js tf.train.adagrad()用法及代碼示例

Tensorflow.js是Google開發的開源庫，用於在瀏覽器或節點環境中運行機器學習模型和深度學習神經網絡。

tf.train.adagrad() 函數用於創建使用自適應梯度算法(adagrad)的 tf.AdagradOptimizer。

用法:

tf.train.adagrad(learningRate).

參數：

learningRate:它指定了自適應梯度下降算法將使用的學習率。
initialAccumulatorValue:它指定累加器的初始值。它必須是積極的。

返回值：它返回一個 tf.adagradOptimizer。

範例1：通過學習係數 x, y 擬合函數 f = (x + y)。

Javascript


// importing tensorflow
import tensorflow as tf
  
const xs = tf.tensor1d([0, 1, 2]);
const ys = tf.tensor1d([1.3, 2.5, 3.7]);
  
const x = tf.scalar(Math.random()).variable();
const y = tf.scalar(Math.random()).variable();
  
// Define a function f(x, y) = x + y.
const f = x => x.add(y);
const loss = (pred, label) => 
    pred.sub(label).square().mean();
  
const learningRate = 0.05;
  
// Create adagrad optimizer
const optimizer = 
  tf.train.adagrad(learningRate);
  
// Train the model.
for (let i = 0; i < 5; i++) {
   optimizer.minimize(() => loss(f(xs), ys));
}
  
// Make predictions.
console.log(
`x:${x.dataSync()}, y:${y.dataSync()}`);
const preds = f(xs).dataSync();
preds.forEach((pred, i) => {
console.log(`x:${i}, pred:${pred}`);
});

輸出

x:0.8561810255050659, y:0.6922483444213867
x:0, pred:0.6922483444213867
x:1, pred:1.6922483444213867
x:2, pred:2.6922483444213867

範例2：通過學習係數 a、b、c 來擬合二次函數。

Javascript


// importing tensorflow
import tensorflow as tf
  
const xs = tf.tensor1d([0, 1, 2, 3]);
const ys = tf.tensor1d([1.1, 5.9, 16.8, 33.9]);
  
const a = tf.scalar(Math.random()).variable();
const b = tf.scalar(Math.random()).variable();
const c = tf.scalar(Math.random()).variable();
  
const f = x => a.mul(
  x.square()).add(b.mul(x)).add(c);
const loss = (pred, label) => 
         pred.sub(label).square().mean();
  
const learningRate = 0.01;
const optimizer = 
      tf.train.adagrad(learningRate);
  
// Train the model.
for (let i = 0; i < 10; i++) {
   optimizer.minimize(() => loss(f(xs), ys));
}
  
// Make predictions.
console.log(
`a:${a.dataSync()}, b:${b.dataSync()}, c:${c.dataSync()}`);
const preds = f(xs).dataSync();
preds.forEach((pred, i) => {
   console.log(`x:${i}, pred:${pred}`);
});

輸出

a:0.3611285388469696, 
b:0.6980878114700317, 
c:0.8787991404533386
x:0, pred:0.8787991404533386
x:1, pred:1.9380154609680176
x:2, pred:3.7194888591766357
x:3, pred:6.223219394683838

參考： https://js.tensorflow.org/api/1.0.0/#train.adagrad

相關用法

注：本文由純淨天空篩選整理自abhinavjain194大神的英文原創作品 Tensorflow.js tf.train.adagrad() Function。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。