当前位置: 首页 > news >正文

实现一个简单的协同过滤推荐算法

题目描述:
协同过滤是推荐系统中的一种常用技术,其基本思想是利用用户之间的相似性或物品之间的相似性来进行推荐。本次面试题要求实现一个基于用户的协同过滤推荐算法。
具体要求:
定义两个类:User 和 Item,分别表示用户和物品。
User 类包含用户ID和用户对各个物品的评分(使用Map<Item, Integer>存储)。
Item 类包含物品ID。
实现一个方法 calculateSimilarity(User user1, User user2),计算两个用户之间的相似度。相似度计算可以使用余弦相似度。
实现一个方法 recommendItems(User targetUser, List<User> otherUsers, int topN),为目标用户推荐N个最可能的物品。推荐逻辑是:找出与目标用户最相似的K个用户,然后根据这些用户的评分来推荐物品。
示例代码:

import java.util.*;class User {private String userId;private Map<String, Integer> ratings;public User(String userId) {this.userId = userId;this.ratings = new HashMap<>();}public void addRating(String item, int rating) {ratings.put(item, rating);}public Map<String, Integer> getRatings() {return ratings;}public static double calculateSimilarity(User user1, User user2) {Map<String, Integer> ratings1 = user1.getRatings();Map<String, Integer> ratings2 = user2.getRatings();double similarity = 0.0;for (String item : ratings1.keySet()) {if (ratings2.containsKey(item)) {similarity += ratings1.get(item) * ratings2.get(item);}}return similarity / (Math.sqrt(ratings1.size() * ratings2.size());}public static List<String> recommendItems(User targetUser, List<User> otherUsers, int topN) {Map<User, Double> similarityScores = new HashMap<>();for (User user : otherUsers) {if (!user.equals(targetUser)) {double similarity = calculateSimilarity(targetUser, user);similarityScores.put(user, similarity);}}List<Map.Entry<User, Double>> sortedUsers = new ArrayList<>(similarityScores.entrySet());sortedUsers.sort((a, b) -> b.getValue().compareTo(a.getValue()));Map<String, Integer> targetRatings = targetUser.getRatings();List<String> recommendedItems = new ArrayList<>();for (int i = 0; i < Math.min(topN, sortedUsers.size()); i++) {User similarUser = sortedUsers.get(i).getKey();Map<String, Integer> similarRatings = similarUser.getRatings();for (String item : similarRatings.keySet()) {if (!targetRatings.containsKey(item)) {recommendedItems.add(item);}}return recommendedItems;}public static void main(String[] args) {// Example usageUser user1 = new User("1");User user2 = new User("2");User user3 = new User("3");user1.addRating("item1", 5);user1.addRating("item2", 3);user2.addRating("item1", 4);user2.addRating("item3", 2);user3.addRating("item2", 5);user3.addRating("item3", 4);List<String> recommendations = recommendItems(user1, Arrays.asList(user2, user3), 2);System.out.println("Recommended items for user1: " + recommendations);}
}


 

http://www.lryc.cn/news/539067.html

相关文章:

  • eNSP防火墙综合实验
  • 操作系统知识(二)
  • 图论:tarjan 算法求解强连通分量
  • Spring中Bean的四种实例化方法
  • 专利申请要求
  • 解锁 JavaScript 异步编程:Promise 链式操作、async/await 与 Promise.all 深度剖析
  • Centos虚拟机扩展磁盘空间
  • 记录一次部署PC端网址全过程
  • 利用 OpenCV 进行棋盘检测与透视变换
  • Java Spring boot 篇:常用注解
  • #渗透测试#批量漏洞挖掘#Apache Log4j反序列化命令执行漏洞
  • 【Linux】Linux 文件系统——关于inode 不足的相关案例
  • k8s集群如何赋权普通用户仅管理指定命名空间资源
  • 工控网络安全介绍 工控网络安全知识题目
  • AIGC(生成式AI)试用 21 -- Python调用deepseek API
  • 跨平台AES/DES加密解密算法【超全】
  • Webpack 基础入门
  • deepseek-v3在阿里云和腾讯云的使用中的差异
  • Mathtype安装入门指南
  • 使用 Apache PDFBox 提取 PDF 中的文本和图像
  • 【js逆向_入门】图灵爬虫练习平台 第四题
  • Redis7——基础篇(三)
  • 深度学习中的知识蒸馏
  • 【Windows软件 - HeidiSQL】导出数据库
  • 苏剑林“闭门造车”之多模态思路浅谈思考
  • 绿联nas docker 安装 rocketmq 队列。亲测可用
  • C++(23):unreachable
  • 初等数论--欧几里得算法
  • 阿里云前端自动化部署流程指南
  • EXCEL解决IF函数“您已为此函数输入太多个参数”的报错