映画-映画類似度ベース推薦リストの作成
バックエンド
モデル
backend/api/online/models.py
...（略）...
class ReclistMoviesMovies(models.Model):
"""映画-映画類似度ベース推薦システムによる推薦リストモデル
Attributes
----------
id : IntegerField
推薦リストID
base_movie : ForeignKey[Movie]
ベース映画
rank : IntegerField
推薦順位
movie : ForeignKey[Movie]
推薦映画
score : FloatField
推薦スコア
"""
id = models.TextField(primary_key=True, max_length=5)
base_movie = models.ForeignKey(Movie, on_delete=models.CASCADE, related_name='base_movie_movies')
rank = models.IntegerField(blank=False, null=False)
movie = models.ForeignKey(Movie, on_delete=models.CASCADE, related_name='movie_movies')
score = models.FloatField()
class Meta:
managed = True
db_table = 'reclist_movies_movies'
def __str__(self):
return '{}:{}:{}({})'.format(self.base_movie.id, self.rank, self.movie.id, self.score)
マイグレーション
(recsys_full) backend$ python manage.py makemigrations online --settings config.settings.development
Migrations for 'online':
api/online/migrations/0005_reclistmoviesmovies.py
+ Create model ReclistMoviesMovies
(recsys_full) backend$ python manage.py migrate --settings config.settings.development
データの登録
offline/data$ psql recsys_full -U postgres -c "\copy reclist_movies_movies (id, base_movie_id, rank, movie_id, score) from 'reclist_movies_movies.csv' with delimiter E'\t' csv header encoding 'UTF8'"
データの確認
recsys_full=# SELECT * FROM reclist_movies_movies;
ビュー
backend/api/online/views.py
from rest_framework.views import APIView
from rest_framework.response import Response
from rest_framework import status
from .models import User, Movie, Rating
from .models import ReclistPopularity, ReclistMoviesMovies # <- ReclistMoviesMoviesを追加
from .mappers import UserMapper, MovieMapper, RatingMapper
from .utils import hash
import uuid
from django.db.models import Prefetch
...（略）...
# ↓追加
class MoviesMoviesMoviesView(APIView):
"""映画-映画類似度ベース推薦システムによる推薦リストビュークラス
"""
def get(self, request, format=None):
"""ベース映画とのアイテム類似度に基づく推薦リストを取得する。
Requests
--------
base_movie_id : int
ベース映画ID
user_id : str
ユーザID
Response
--------
movies : json
映画リスト
"""
# ユーザ認証
user_id = request.GET.get('user_id') if 'user_id' in request.GET else None
# リクエストパラメタの取得
base_movie_id = request.GET.get('base_movie_id')
# オブジェクトの取得
reclist = []
if user_id:
reclist = ReclistMoviesMovies.objects.filter(base_movie_id=base_movie_id)\
.select_related('movie')\
.prefetch_related('movie__genres')\
.prefetch_related(
Prefetch(
'movie__movie_ratings',
queryset=Rating.objects.filter(user_id=user_id),
to_attr='user_ratings',
)
)
else:
reclist = ReclistMoviesMovies.objects.filter(base_movie_id=base_movie_id)\
.select_related('movie')\
.prefetch_related('movie__genres')
# レスポンス
movies = [rec.movie for rec in reclist]
movies_dict = [MovieMapper(movie).as_dict(user_id) for movie in movies]
data = {
'movies': movies_dict,
}
return Response(data, status.HTTP_200_OK)
# ↑追加
URLマッピング
backend/api/online/urls.py
from django.urls import path
from . import views
urlpatterns = [
path('users/', views.UsersView.as_view()),
path('movies/', views.MoviesView.as_view()),
path('movies/<int:id>/', views.MovieView.as_view()),
path('ratings/', views.RatingsView.as_view()),
path('movies_popularity/', views.MoviesPopularityView.as_view()),
path('movies_movies_movies/', views.MoviesMoviesMoviesView.as_view()), # <- 追加
]
ブラウザで下記それぞれのURLにアクセスしてください。
- http://localhost:8000/api/online/movies_movies_movies/?base_movie_id=1
- http://localhost:8000/api/online/movies_movies_movies/?base_movie_id=1&user_id=【ユーザID】
base_movie_idを変えてアクセスすると、ベース映画ごとに類似映画が推薦リストとして表示されます。
フロントエンド
API
frontend/src/services/movies/getMoviesMoviesMovies.ts
import { ApiContext, Movie, User } from '@/types/data';
import { fetcher } from '@/utils';
const context: ApiContext = {
apiRootUrl: process.env.NEXT_PUBLIC_API_BASE_URL,
};
/**
* 映画-映画推薦システムによる映画リスト取得API
* @param baseMovie - ベース映画
* @param user - ユーザ
* @returns 映画リスト
*/
const getMoviesMoviesMovies = async (
baseMovie: Movie,
user?: User
): Promise<{ movies: Movie[] }> => {
const userParam = user ? `&user_id=${user.id}` : '';
return await fetcher(
`${context.apiRootUrl?.replace(/\/$/g, '')}/movies_movies_movies/?base_movie_id=${baseMovie.id}${userParam}`,
{
headers: {
Accept: 'application/json',
'Content-Type': 'application/json',
},
cache: 'no-store',
}
);
};
export default getMoviesMoviesMovies;
コンポーネント
frontend/src/app/components/movie/MovieListMoviesMovies.tsx
import { Movie, User } from '@/types/data';
import React from 'react';
import MovieList from './MovieList';
import getMoviesMoviesMovies from '@/services/movies/getMoviesMoviesMovies';
type Props = {
baseMovie: Movie;
perPage: number;
user: User;
};
const MovieListMoviesMovies = async (props: Props) => {
const phrase = 'この映画が好きな人はこんな映画も好んでいます';
const { movies } = await getMoviesMoviesMovies(props.baseMovie, props.user);
return (
<>
<MovieList phrase={phrase} movies={movies} perPage={props.perPage} user={props.user} />
</>
);
};
export default MovieListMoviesMovies;
ページ
frontend/src/app/movies/[id]/page.tsx
import MovieCardDetail from '@/app/components/movie/MovieCardDetail';
import MovieListMoviesMovies from '@/app/components/movie/MovieListMoviesMovies'; // <- 追加
import { auth } from '@/auth';
import getMovie from '@/services/movies/getMovie';
import connectUser from '@/services/users/connectUser';
import getUser from '@/services/users/getUser';
const PER_PAGE = 5; // <- 追加
const Movie = async ({ params }: { params: Promise<{ id: number }> }) => {
await connectUser();
const session = await auth();
const user = session ? await getUser(session?.user?.email!) : null;
const { id } = await params;
const movieId = id;
const { movie } = await getMovie(movieId, user!);
return (
<>
<MovieCardDetail movie={movie} user={user!} />
<MovieListMoviesMovies baseMovie={movie} user={user!} perPage={PER_PAGE} /> {/* <- 追加 */}
</>
);
};
export default Movie;
ブラウザで下記URLにアクセスしてください。
対象映画の類似映画が推薦リストとして提示されます。