Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- DataModel model = new ReloadFromJDBCDataModel(new MySQLJDBCDataModel(dataSource, "ps_product_comment", "id_customer", "id_product", "grade", null));
- UserSimilarity userSimilarity = new PearsonCorrelationSimilarity(model);
- UserNeighborhood neighborhood =
- new ThresholdUserNeighborhood(0.0001, userSimilarity, model);
- //tutaj wyppisuję sąsiadów danego usera
- for(Long user : neighborhood.getUserNeighborhood(userId)){
- System.out.println("UserNeighborhood for user: " + user);
- }
- GenericUserBasedRecommender recommender =
- new GenericUserBasedRecommender(model, neighborhood, userSimilarity);
- //tutaj wyppisuję preferencje danego usera
- for(int i = 1; i < 400; i++){
- if(recommender.estimatePreference(userId, i)!= Float.NaN)
- System.out.println("Preference for item "+ i +": " + recommender.estimatePreference(userId, i));
- }
- List<RecommendedItem> recommendations =
- recommender.recommend(userId, 4);
- System.out.println("userId: " + userId);
- //a tu rekomendacje
- System.out.println("recommendations: " + recommendations.toString());
- //jak nie ma sąsiadów, to nie ma rekomendacji
- List<ScorpionRecommendation> list = new ArrayList<>();
- for(RecommendedItem recommendation : recommendations){
- list.add(db.getRecommendation(recommendation.getItemID()));
- }
- System.out.println("list: " + list.toString());
- return list;
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement