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)