Mahout - 运行 als-wr 推荐器评估器

Question

我正在尝试使用 mahout 在我的输入中获得 als-wr 算法的精度和召回率。

问题是我发现使用 als-wr 算法的唯一方法是来自 mahout 命令行 - https://mahout.apache.org/users/recommender/intro-als-hadoop.html

我在网络上没有看到任何在代码中使用此算法的示例。

非常感谢任何在 scala/java 中使用评估器运行它的示例或帮助。

Answer 1

    RecommenderIRStatsEvaluator user_eval = new GenericRecommenderIRStatsEvaluator();
    RecommenderBuilder user_rb = new RecommenderBuilder() 
    {
    @Override
        public Recommender buildRecommender(DataModel model) throws TasteException 
        {
            UserSimilarity user_ll_sim = new LogLikelihoodSimilarity(model);
            UserNeighborhood user_ll_nbhd = new NearestNUserNeighborhood(5, user_ll_sim, model);
            return new GenericUserBasedRecommender(model, user_ll_nbhd, user_ll_sim);
        }
    };

    IRStatistics user_ll_stats = user_eval.evaluate(user_ll_rb, null, model, null, 2, GenericRecommenderIRStatsEvaluator.CHOOSE_THRESHOLD, 0.8);
    System.out.println(user_ll_stats.getPrecision()+" "+user_ll_stats.getRecall());

将基于用户的对数似然模型替换为 SVD 模型。