def dataset(self): dl = Distribution([1, 0], [ self.probiablity, ]) v = dl.generator(len(self.dset)) m = [] for i in range(len(self.dset)): if v[i] == 1: m.append(self.dset) return m