def __init__()

in 01-byoc/code/dataset.py [0:0]

• 12 lines of code
• 2 McCabe index (conditional complexity)

    def __init__(self, params):
        """Init SoundDataset with params
        Args:
            params (class): all arguments parsed from argparse
            train (bool): train or val dataset
        """
        self.params = params
        self.csvfile = params.csv_path
        self.data_dir = params.data_dir
        self.normalize = self.params.normalize
        self.preload = self.params.preload

        self.X, self.Y, self.filenames = self.read_data(self.csvfile)
        if self.preload:
            self.X = self.convert_to_spec(self.X)
            self.shape = self.get_shape(self.X[0])
        else:
            self.shape = self.get_shape(self.preprocessing(self.X[0][0], self.X[0][1]))