(self, path)
| 34 | self.loadAllFiles(self.path+"/") |
| 35 | |
| 36 | def loadAllFiles(self, path): |
| 37 | Log("Loading files...") |
| 38 | |
| 39 | if self.FILES_AMOUNT<=1: |
| 40 | self.loadFile(path+"train", False) |
| 41 | self.loadFile(path+"test", True) |
| 42 | else: |
| 43 | for i in range(self.FILES_AMOUNT): |
| 44 | file_name = path+"data_batch_"+str(i+1) |
| 45 | self.loadFile(file_name) |
| 46 | SetState(self.id, i/self.FILES_AMOUNT) |
| 47 | |
| 48 | file_name = path+"test_batch" |
| 49 | self.loadFile(path+"test_batch", True) |
| 50 | |
| 51 | def loadFile(self, file, isTest=False): |
| 52 | fo = open(file, 'rb') |