def calTFIDFDIFF(g1, g2, genres_tag_dict): g1_movie_tag_dict = movieTag(genres_tag_dict, g1) g2_movie_tag_dict = movieTag(genres_tag_dict, g2) g1_g2_movie_tag_dict = mergeG1G2(g1_movie_tag_dict, g2_movie_tag_dict) genres_tag_movie = mapMovieToGenre(genres_tag_dict, g1_g2_movie_tag_dict) g1_TFIDF = putil.calDocTFIDF(genres_tag_dict, g1, 'genre', movie=True, movie_tag_dict=genres_tag_movie) g2_TFIDF = putil.calDocTFIDF(genres_tag_dict, g2, 'genre', movie=True, movie_tag_dict=g1_g2_movie_tag_dict) return g1_TFIDF, g2_TFIDF
def printUser(mlratings, mltags, tag_name_dict, userid, model): user_tag_dict = prepareData(mlratings, mltags) res = {} if model == 'TF': res = putil.calDocTagTF(user_tag_dict, userid) else: res = putil.calDocTFIDF(user_tag_dict, userid, 'user') putil.print_result(model, 'user', userid, tag_name_dict, res)
def printGenre(genres_movie, mltags, tag_name_dict, genreid, model): genre_tag_dict = prepareData(genres_movie, mltags) res = {} if model == 'TF': res = putil.calDocTagTF(genre_tag_dict, genreid) else: res = putil.calDocTFIDF(genre_tag_dict, genreid, 'genre') putil.print_result(model, 'genre', genreid, tag_name_dict, res)
def printActor(movie_actor, mltags, tag_name_dict, actorid,model): actor_tag_dict = prepareData(movie_actor, mltags) res = {} if model == 'TF': res = putil.calDocTagTF(actor_tag_dict, actorid, isactor = True) else: res = putil.calDocTFIDF(actor_tag_dict, actorid, 'actor') putil.print_result(model, 'actor', actorid, tag_name_dict, res)