02 Bar _ Recommandation de film (basée sur le contenu) Portrait de l'utilisateur

Recommandation de films basés sur le contenu：Portrait de l'utilisateur

Étapes de construction du portrait de l'utilisateur：

• Selon l'historique de notation de l'utilisateur,Combinaison de portraits d'objets,.Retourner l'étiquette de l'image du film avec l'enregistrement de l'image à l'utilisateur comme étiquette initiale
• En comptant le nombre de fois que l'utilisateur regarde l'étiquette d'ombre,Calculer les valeurs de poids pour chaque étiquette initiale de l'utilisateur,Trier et sélectionnerTOP-NÉtiquette d'image finale de l'utilisateur

Création de portraits d'utilisateurs

import pandas as pd
import numpy as np
from gensim.models import TfidfModel

from functools import reduce
import collections

from pprint import pprint

# ......

''' user profileCréation de portraits： 1. Extraire la liste de visualisation de l'utilisateur 2. Correspond aux mots clés de l'utilisateur en fonction de la liste de visualisation et de l'image de l'article,Et compter la fréquence des mots 3. Trier par fréquence de mots,Réserve maximaleTOP-kUn mot.,Ici.KSet to100,Étiquette en tant qu'utilisateur '''

def create_user_profile():
    watch_record = pd.read_csv("datasets/ml-latest-small/ratings.csv", usecols=range(2), dtype={
    "userId":np.int32, "movieId": np.int32})

    watch_record = watch_record.groupby("userId").agg(list)
    # print(watch_record)

    movie_dataset = get_movie_dataset()
    movie_profile = create_movie_profile(movie_dataset)

    user_profile = {
    }
    for uid, mids in watch_record.itertuples():
        record_movie_prifole = movie_profile.loc[list(mids)]
        counter = collections.Counter(reduce(lambda x, y: list(x)+list(y), record_movie_prifole["profile"].values))
        # Mots d'intérêt
        interest_words = counter.most_common(50)
        maxcount = interest_words[0][1]
        interest_words = [(w,round(c/maxcount, 4)) for w,c in interest_words]
        user_profile[uid] = interest_words

    return user_profile

user_profile = create_user_profile()
pprint(user_profile)