def get_musics_from_file(file_name, list_of_all_tags): musics = [] for line in get_content_from_file(file_name): list_from_line = line.strip().split() music = Music(list_from_line[0], list_from_line[1:], list_of_all_tags) musics.append(music) return musics
def get_set_of_tags_from_file(file_name): content = get_content_from_file(file_name) tags = set() for line in content: tags_from_line = line.strip().split()[1:] tags = tags.union(set(tags_from_line)) return tags
def get_centroids_from_file(musics, k, file_name): centroids = [] for line in get_content_from_file(file_name): try: centroids.append(Point(musics[int(line)].point.vector)) except IndexError: exit('Specified observation ID doesn\'t exist') if len(centroids) != k: exit('Specified K differs from number of cluster in file') return centroids