From fcd2c83048aab529f1db80f7fb77c22ba59b7e1f Mon Sep 17 00:00:00 2001 From: "@t2652009480" <2652009480@qq.com> Date: Mon, 26 Aug 2024 16:25:16 +0800 Subject: [PATCH] =?UTF-8?q?=E6=95=B0=E5=AD=97=E8=90=A5=E9=94=80=E5=AE=9E?= =?UTF-8?q?=E8=AE=AD=E7=AE=97=E6=B3=95=E7=AC=AC=E5=9B=9B=E8=BD=AE=E4=BF=AE?= =?UTF-8?q?=E6=94=B9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../stu/StuDigitalMarketingModelController.java | 6 +++--- .../service/impl/StuDigitalMarketingModelServiceImpl.java | 4 ++-- .../java/com/sztzjy/marketing/util/algorithm/Apriori.java | 8 ++++---- 3 files changed, 9 insertions(+), 9 deletions(-) diff --git a/src/main/java/com/sztzjy/marketing/controller/stu/StuDigitalMarketingModelController.java b/src/main/java/com/sztzjy/marketing/controller/stu/StuDigitalMarketingModelController.java index 63a08ec..e85bdba 100644 --- a/src/main/java/com/sztzjy/marketing/controller/stu/StuDigitalMarketingModelController.java +++ b/src/main/java/com/sztzjy/marketing/controller/stu/StuDigitalMarketingModelController.java @@ -191,13 +191,13 @@ public class StuDigitalMarketingModelController { List strings = convertToDelimitedStringList(deduplicatedDataList); //初始化事务数据库、项目集、候选集再进行剪枝 - Apriori.init(strings,support); + Apriori.init(strings,confidence); //迭代求出最终的候选频繁集 - Apriori.iteration(Apriori.C,Apriori.L,support); + Apriori.iteration(Apriori.C,Apriori.L,confidence); //根据最终的关联集,根据公式计算出各个关联事件 - List connection = Apriori.connection(confidence); + List connection = Apriori.connection(support); return new ResultEntity(HttpStatus.OK,"成功",connection); diff --git a/src/main/java/com/sztzjy/marketing/service/impl/StuDigitalMarketingModelServiceImpl.java b/src/main/java/com/sztzjy/marketing/service/impl/StuDigitalMarketingModelServiceImpl.java index 67b12a0..8ad6ada 100644 --- a/src/main/java/com/sztzjy/marketing/service/impl/StuDigitalMarketingModelServiceImpl.java +++ b/src/main/java/com/sztzjy/marketing/service/impl/StuDigitalMarketingModelServiceImpl.java @@ -57,7 +57,7 @@ public class StuDigitalMarketingModelServiceImpl implements StuDigitalMarketingM List list=stuTableNameMapper.getTableName(userId); if(list.isEmpty()){ //设置默认值 - String[] arr={"客户细分数据表","销售预测数据表","购物车数据表","用户评论表"}; + String[] arr={"客户细分数据表","销售预测数据表","购物车数据表"}; StuTableName stuTableName=new StuTableName(); for (int i = 0; i < arr.length; i++) { @@ -306,7 +306,7 @@ public class StuDigitalMarketingModelServiceImpl implements StuDigitalMarketingM if(tableName.equals(Constant.KEHUXISHUJU)){ // list=indicatorsMapper.getYHSXB(); - list.add("id"); +// list.add("id"); list.add("gender"); list.add("age"); list.add("annual_income"); diff --git a/src/main/java/com/sztzjy/marketing/util/algorithm/Apriori.java b/src/main/java/com/sztzjy/marketing/util/algorithm/Apriori.java index b995df7..a70aaf4 100644 --- a/src/main/java/com/sztzjy/marketing/util/algorithm/Apriori.java +++ b/src/main/java/com/sztzjy/marketing/util/algorithm/Apriori.java @@ -162,13 +162,13 @@ public class Apriori { } str = C.toString(); -// System.out.println("候选"+t+"项集:C: \n"+C); + System.out.println("候选"+t+"项集:C: \n"+C); // 二、剪枝步 pruning(C, L,min_support); -// System.out.println("频繁"+t+"项集:L: \n"+L+"\n"); + System.out.println("频繁"+t+"项集:L: \n"+L+"\n"); str = L.toString(); - //System.out.println("==="); + System.out.println("==="); L_ALL.putAll(L); } return L_temp; @@ -252,7 +252,7 @@ public class Apriori { if (confident > min_confident) {// 如果事件的置信度大于最小置信度 //封装对象返回 - DecimalFormat df = new DecimalFormat("#.0"); + DecimalFormat df = new DecimalFormat("#.00"); AssociationRulesDTO associationRulesDTO=new AssociationRulesDTO(); associationRulesDTO.setCorrelation(item_pre); associationRulesDTO.setAssociated(item_post);