1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27
| public class EvaluateRecommender {
static Scanner clavier = new Scanner (System.in);
public static void main(String[] args) throws Exception{
DataModel model = new FileDataModel(new File("data/ml-100k/ua.base"));
RecommenderBuilder BuildRecom = new MyRecommenderBuilder();
RecommenderEvaluator evaluation = new AverageAbsoluteDifferenceRecommenderEvaluator();
double score = evaluation.evaluate(BuildRecom, null, model, 0.9, 1.0);
System.out.println(score);
}
}
public class MyRecommenderBuilder implements RecommenderBuilder {
public Recommender BuildRecommender(DataModel dataModel) throws TasteException{
UserSimilarity similarity = new PearsonCorrelationSimilarity(dataModel);
UserNeighborhood neighborhood = new ThresholdUserNeighborhood(0.1,similarity, dataModel);
// NearestNUserNeighborhood neighborhood = new NearestNUserNeighborhood (0.1,userSimilarity, dataModel);
return new GenericUserBasedRecommender(dataModel,neighborhood,similarity);
}
} |
Partager