nn.py 文件源码

def __init__(self, d, batchsize, n_train_epoch, n_val_epoch, n_units, gpu):
        self.d = d
        self.batchsize = batchsize
        self.n_train_epoch = n_train_epoch
        self.n_val_epoch = n_val_epoch
        self.n_units = n_units
        self.optimizer = optimizers.Adam()
        self.model = L.Classifier(MLP(self.d, self.n_units, 2))
        if gpu:
            self.model.to_gpu(0)
        self.optimizer.setup(self.model)