def collect_data(directory):
    samples, classes = list(), list()
    for i, subdir in enumerate(["normal", "murmur", "extrasystole"]):
        for filepath in glob(os.path.join(directory, subdir, "*.wav")):
            features = extract_features(filepath)
            ext_features = np.hstack(features)
            features = np.vstack([features, ext_features])
            print(features)
            return
            samples.append(extract_features(filepath))
            classes.append(i)
    return samples, classes


if __name__ == "__main__":
    samples, classes = collect_data("data")
    print(samples, classes)