def reset_dropout()

in data/dataset.py [0:0]

• 16 lines of code
• 4 McCabe index (conditional complexity)

    def reset_dropout(self):
        self.sample_indices = list(range(len(self.data)))
        self.sample_pairs = self.train_pairs

        shuffled_ind = np.random.permutation(len(self.train_pairs))
        n_pairs = int((1 - self.pair_dropout) * len(self.train_pairs))
        self.sample_pairs = [
            self.train_pairs[pi] for pi in shuffled_ind[:n_pairs]
        ]
        print('Using {} pairs out of {} pairs right now'.format(
            n_pairs, len(self.train_pairs)))
        self.sample_indices = [
            i for i in range(len(self.data))
            if (self.data[i][1], self.data[i][2]) in self.sample_pairs
        ]
        print('Using {} images out of {} images right now'.format(
            len(self.sample_indices), len(self.data)))