def __init__(self, n_in, n_mid, n_out, batchsize, gpu=-1):
DLBases.__init__(self, batchsize, gpu)
# モデル構築
self.model = ffnn.FFNN(n_in, n_mid, n_out)
# GPU設定
if gpu >= 0:
chainer.cuda.get_device_from_id(gpu).use()
cuda.check_cuda_available()
self.model.to_gpu()
# 最適化手法をAdamに設定
self.optimizer = optimizers.Adam()
self.optimizer.setup(self.model)
def __init__(self, n_in, n_mid, batchsize, gpu=-1, window_size=1):
DLBases.__init__(self, batchsize=batchsize, gpu=gpu)
# モデル構築
self.model = ffnn.FFNN(n_in, n_mid, 5)
# モデルに対するGPU設定
if gpu >= 0:
chainer.cuda.get_device_from_id(gpu).use()
cuda.check_cuda_available()
self.model.to_gpu()
# 最適化手法をAdamに設定
self.optimizer = optimizers.Adam()
self.optimizer.setup(self.model)
# パラメータ保持
self.window_size = window_size