physbo.opt.adam module

class physbo.opt.adam.adam(params, grad, options={})

ベースクラス: object

Optimizer of f(x) with the adam method

params

current input, x

Type

numpy.ndarray

nparams

dimension

Type

int

grad

gradient function, g(x) = f'(x)

Type

function

m

Type

numpy.ndarray

v

Type

numpy.ndarray

epoch

the number of update already done

Type

int

max_epoch

the maximum number of update

Type

int

alpha

Type

float

beta

Type

float

gamma

Type

float

epsilon

Type

float

run(*args, **kwargs)

set_params(params)

update(params, *args, **kwargs)

calculates the updates of params

パラメータ

• params (numpy.ndarray) -- input

• args -- will be passed to self.grad

• kwargs -- will be passed to self.grad

戻り値

update of params

戻り値の型

numpy.ndarray