Method checkDataNumber

Classification.py:251–259

(self, trainNum, validationNum)

        return validation_labels

    def checkDataNumber(self, trainNum, validationNum):
        if trainNum % self.batch_size != 0:
            raise ValueError('trainNum:{} can not exact div batch_size={},please delete {} sample(s) in manual'
                             .format(trainNum, self.batch_size,
                                     trainNum - trainNum // self.batch_size * self.batch_size))
        if validationNum % self.batch_size != 0:
            raise ValueError('validationNum:{} can not exact div batch_size={},please delete {} sample(s) in manual'
                             .format(validationNum, self.batch_size,
                                     validationNum - validationNum // self.batch_size * self.batch_size))

    def run(self):
        fileInfo = self.readTotalFileNum()