コード例 #1
0
ファイル: dataset.py プロジェクト: echogecho/cifar-ten
 def __init__(self):
     batch1 = cifar10.batch1()
     batch2 = cifar10.batch2()
     batch3 = cifar10.batch3()
     x = numpy.vstack([batch1['data'], batch2['data'], batch3['data']])
     y = numpy.concatenate([batch1['labels'], batch2['labels'], batch3['labels']])
     self.train = x, y
     self.valid = xy(cifar10.batch4())
     self.test = xy(cifar10.batch5())
コード例 #2
0
ファイル: dataset.py プロジェクト: echogecho/cifar-ten
 def __init__(self):
     self.train = xy(cifar10.batch1())
     self.valid = xy(cifar10.batch4())
     self.test = xy(cifar10.batch5())