tools.py 文件源码

def __init__(self,
               name,
               number_of_classes,
               number_of_transformations,
               loaded_size,
               desired_size,
               max_size=None):
    loaded = np.loadtxt(name)
    if max_size is not None:
      subset = np.random.choice(loaded.shape[0], max_size, replace=False)
      loaded = loaded[subset, :]
    padded_x = self._pad(loaded[:, :-1], loaded_size, desired_size)
    self._x = self._transform(padded_x, number_of_transformations)
    self._y = self._int_labels_to_one_hot(loaded[:, -1], number_of_classes)
    self._completed_epochs = -1
    self._new_epoch = False
    self._start_new_epoch()