mahout学习(二)

下载测试数据:http://grouplens.org/datasets/movielens/   ml-100k.zip

public class TMahout02 {

    public static void main(String[] args) throws IOException, TasteException {

        //-配置并评估一个推荐程序-//
        RandomUtils.useTestSeed();
        DataModel model = new FileDataModel(new File("path/ua.base"));
        RecommenderEvaluator evaluator = new AverageAbsoluteDifferenceRecommenderEvaluator();
        RecommenderBuilder builder = new RecommenderBuilder() {
            @Override
            public Recommender buildRecommender(DataModel dataModel) throws TasteException {
                UserSimilarity similarity = new PearsonCorrelationSimilarity(dataModel);
                //UserNeighborhood neighborhood = new ThresholdUserNeighborhood(0.1, similarity, dataModel);

                UserNeighborhood neighborhood =
                        new NearestNUserNeighborhood(2, similarity, dataModel);

                return new GenericUserBasedRecommender(dataModel, neighborhood, similarity);
            }
        };
        double result = evaluator.evaluate(builder, null, model, 0.9, 1.0);
        System.out.println(result);

    }

}

 

posted @ 2014-09-02 12:04  spacewalk  阅读(173)  评论(0)    收藏  举报