import csv import pickle grafo = Grafo() table_movies = {} with open('datasets/movies.csv', 'r', encoding="UTF-8") as csvmovies: for line in csvmovies: data = line.split(',') id_movie = data[0] name_movie = data[1] table_movies[id_movie] = name_movie.strip() with open('datasets/ratings.csv', 'r', encoding="UTF8") as csvratings: for line in csvratings: data = line.split(',') user = data[0] id_movie = data[1] rated_movie = table_movies[id_movie] rating = float(data[2]) grafo.adicionar_vertice(user) grafo.adicionar_vertice(rated_movie) grafo.adicionar_aresta(user, rated_movie, rating) grafo.adicionar_aresta(rated_movie, user, rating) dados = grafo.grafo with open('dados_grafo.pkl', mode='wb') as dados_grafo: pickle.dump(dados, dados_grafo)