nn.py 文件源码

def __init__(self, d, batchsize, n_train_epoch, n_val_epoch, n_units):
        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.model = L.Classifier(MLP(self.d, self.n_units, 2))
        self.model.o = optimizers.Adam()
        self.model.o.setup(self.model)