BS1065-基于数据分析+推荐算法+数据可视化的特征新闻推荐系统

public TFIDF(List<String> documents) {
    wordCount = new HashMap<>();
    idf = new HashMap<>();
    tfidf = new HashMap<>();

    //}
    }

    for (String word : wordCount.keySet()) {
        double idfValue = Math.log10((double) documents.size() / wordCount.get(word));
        idf.put(word, idfValue);
    }

    // 计算TF-IDF
    for (String document : documents) {
        Map<String, Double> documentTfIdf = new HashMap<>();
        String[] words = document.split(" ");
        for (String word : words) {
            double tfValue = (double) wordCount.get(word) / words.length;
            double tfidfValue = tfValue * idf.get(word);
            documentTfIdf.put tfidf;
}

private Page<WhzyInfoEntity> findSimilarUserBook(Map<String, Object> params) {

        String beanName = (String) params.get("paramKey");
        String wid = (String) params.get("wid");
        Wrapper<WhzyInfoEntity> qw = new EntityWrapper<WhzyInfoEntity>()
                .setSqlSelect("pid", "name", "xx", "zz", "zy", "version", "pn", "wn", "bsize", "paper", "mark", "tjrq", "ptime", "imgw", "content").eq("status", 1);
        String matchText;
        if (StringUtils.isNotEmpty(wid) && !"null".equalsIgnoreCase(wid)) {//基于文章内容推荐
            WhzyInfoEntity entity = selectBookInfo(wid);
            if (null == entity) return null;
            matchText = entity.getContent();
            qw.ne("pid", wid);
        } else {//基于关键词推荐
            matchText = beanName;
            qw.like(StringUtils.isNotBlank(matchText), "name", matchText);
        }
        String finalMatchText = matchText;
        if (StringUtils.isEmpty(finalMatchText)) return null;
        List<WhzyInfoEntity> books = this.selectList(qw);
        List<CompletableFuture<WhzyInfoEntity>> collect = books.stream().map(item -> CompletableFuture.supplyAsync(() -> {
            double similarity = Cosine.getSimilarity(finalMatchText, item.getContent());
            item.setTjd(Double.valueOf(String.format("%.3f", similarity)));
            return item;
        }, executorPool)).collect(Collectors.toList());
        List<WhzyInfoEntity> resultList = collect.stream().map(c -> c.join()).sorted((o1, o2) -> {
            Double tjd1 = o1.getTjd();
            Double tjd2 = o2.getTjd();
            return tjd2.compareTo(tjd1);
        }).filter(it -> it.getTjd() > 0).collect(Collectors.toList());

        Page<WhzyInfoEntity> page = new Page<>();
        Integer limit = Integer.valueOf((String) params.get("limit"));
        Integer cpage = Integer.valueOf((String) params.get("page"));
        List<List<WhzyInfoEntity>> blists = ListCF.subList(resultList, limit);
        page.setRecords(cpage > blists.size() ? Lists.newArrayList() : blists.get(cpage - 1));
        page.setTotal(resultList.size());
        page.setSize(limit);
        page.setCurrent(cpage);
        if (StringUtils.isNotEmpty(beanName)) {
            page.getRecords().stream().forEach(item -> item.setName(item.getName().replace(beanName, "<span style='color: #ef0000'>" + beanName + "</span>")));
        }
        return page;
    }