def getSongLabels(self):
'''
This is used externally to get all of
the song labels after the songs have been labeled. It returns
the dictionary of song labels
INPUT: NONE
OUTPUT: dictionary, keys = song names, values = song labels
'''
df = pd.read_csv(Globals.getLabelsDir()+"\\"+"labels.csv")
return dict(df.values)
def write_pre_labels(self):
'''
This loads all of the song names in the song directory
and creates a csv called pre_labels where each song will be
manually labeled. the resulting file should be saved as labels.csv
INPUT: NONE
OUTPUT: NONE
'''
song_dir = Globals.getSongDir()
file_list = [f for f in listdir(song_dir) if isfile(join(song_dir, f))]
read_count = 0
song_names = []
for file_name in file_list:
if ".wav" not in file_name:
break
read_count += 1
song_names.append(file_name)
self.write_pre_labels_to_csv(song_names, Globals.getLabelsDir())
