def read_data()

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

• 26 lines of code
• 8 McCabe index (conditional complexity)

    def read_data(self, csvfile):
        df = pd.read_csv(csvfile)
        data, label, filenames = [], [], []
        print("reading wav files...")
        for i in tqdm(range(len(df))):
            row = df.iloc[i]
            path = os.path.join(self.data_dir, row.Filename + ".wav")
            wav_data, sr = wav_read(path)
            assert wav_data.dtype == np.int16
            data.append((wav_data, sr))
            lb = None
            if row.Barking == 1:
                lb = 0
            elif row.Howling == 1:
                lb = 1
            elif row.Crying == 1:
                lb = 2
            elif row.COSmoke == 1:
                lb = 3
            elif row.GlassBreaking == 1:
                lb = 4
            elif row.Other == 1:
                lb = 5
            label.append(lb)
            filenames.append(path)
        return data, label, filenames