概要
製品説明文書をBERTで埋め込み、内積類似度に基づいて類似製品を推薦するコンテンツベース推薦システム(CBF)を実装しました。
主要成果
- BERTモデルを活用した製品説明文書埋め込みパイプライン構築
- 内積類似度ベースの類似製品推薦アルゴリズム実装
技術スタック
- Model: BERT (HuggingFace Transformers)
- Framework: PyTorch
製品説明文書をBERTで埋め込み、内積類似度に基づいて類似製品を推薦するコンテンツベース推薦システム(CBF)を実装しました。