Example #1
0
def get_clean_graph_generator_memory_repos():
    artists_ngram_repo = MemoryEntityNgrams(base_entity_uri=base_entities_URI,
                                            type_of_entity_collection=ARTIST_COLLECTION)
    songs_ngram_repo = MemoryEntityNgrams(base_entity_uri=base_entities_URI,
                                          type_of_entity_collection=SONG_COLLECTION)
    entity_counter_repo = MemoryEntityCounter()

    songs_ngram_repo.reset_collection()
    artists_ngram_repo.reset_collection()
    entity_counter_repo.reset_count()

    return GraphGenerator(repo_songs=songs_ngram_repo,
                          repo_artist=artists_ngram_repo,
                          repo_counter=entity_counter_repo)