見出し画像

【プログラミング】SGD(確率的勾配降下法)


問題

以下は、確率的勾配降下法(SGD) を用いてパラメータを更新するクラスの一部である。

SGDでは、損失関数の勾配に学習率を掛けた値を現在のパラメータから減算することで更新を行う。

次のコードの空欄に入る最も適切な処理を選べ。


class SGD(object):
    def __init__(self, lr):
        self.lr = lr

    def update(self, params, grads):
        for key in params.keys():
            params[key]  (あ)

選択肢

(a)
-= self.lr * grads[key]

(b)
+= self.lr * grads[key]

(c)
-= self.lr * self.lr * grads[key]

(d)
+= self.lr * self.lr * grads[key]

回答

(a)


いいなと思ったら応援しよう!