liutongsz/fibo-rule
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

机器学习模型上传文件问题修改

Browse Source
This commit is contained in:
刘通
2025-02-11 16:11:12 +08:00
parent ca6466d59a
commit f9f00ea4bd
3 changed files with 17 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
ddp/ddp-common/ddp-service/src/main/java/com/fibo/ddp/common/service/strategyx/aimodel/PMMLExecutor/impl/PMMLExecutorRFImpl.java
View File

@@ -24,6 +24,7 @@ public class PMMLExecutorRFImpl implements PMMLExecutor {
@Override @Override
public Evaluator loadPmml(String filePath) { public Evaluator loadPmml(String filePath) {
// String filePath = "D:\\models\\model_RF.pmml"; // String filePath = "D:\\models\\model_RF.pmml";
Evaluator modelEvaluator = null;
PMML pmml = new PMML(); PMML pmml = new PMML();
InputStream inputStream = null; InputStream inputStream = null;
@@ -52,10 +53,15 @@ public class PMMLExecutorRFImpl implements PMMLExecutor {
} }
} }
/*ModelEvaluatorFactory modelEvaluatorFactory = ModelEvaluatorFactory.newInstance();
Evaluator evaluator = modelEvaluatorFactory.newModelEvaluator(pmml,pmml.getModels().get(0));
pmml = null;*/
ModelEvaluatorBuilder modelEvaluatorBuilder = new ModelEvaluatorBuilder(pmml);
ModelEvaluatorFactory modelEvaluatorFactory = ModelEvaluatorFactory.newInstance(); ModelEvaluatorFactory modelEvaluatorFactory = ModelEvaluatorFactory.newInstance();
Evaluator evaluator = modelEvaluatorFactory.newModelEvaluator(pmml); modelEvaluatorBuilder.setModelEvaluatorFactory(modelEvaluatorFactory);
pmml = null; modelEvaluator = modelEvaluatorBuilder.build();
return evaluator; modelEvaluator.verify();
return modelEvaluator;
} }
@Override @Override

2
ddp/ddp-common/ddp-service/src/main/java/com/fibo/ddp/common/service/strategyx/listlibrary/impl/ListDbServiceImp.java
View File

@@ -231,7 +231,7 @@ public class ListDbServiceImp extends ServiceImpl<ListDbMapper, ListDb> implemen
ListDb listDb = listDbMapper.selectById(id); ListDb listDb = listDbMapper.selectById(id);
String tableName = "organ" + "_" + listDb.getOrganId() + "_" + listDb.getListType() + "_" + id; String tableName = "organ" + "_" + listDb.getOrganId() + "_" + listDb.getListType() + "_" + id;
// 插入多行数据 insertOne into user_info (user_account,user_name,user_age,user_class) values ('00001', '张三 ','20','计算机系'), ('00002', '李四','19','计算机系'); // 插入多行数据 insertOne into user_info (user_account,user_name,user_age,user_class) values ('00001', '张三 ','20','计算机系'), ('00002', '李四','19','计算机系');
String sqlStr = "insertOne into " + tableName + "(t0, t1, t2, t3, t4, t5, t6, t7, t8, t9, t10, t11, t12, t13, t14, t15, t16, t17, t18, t19, user_id) values "; String sqlStr = "insert into " + tableName + "(t0, t1, t2, t3, t4, t5, t6, t7, t8, t9, t10, t11, t12, t13, t14, t15, t16, t17, t18, t19, user_id) values ";
// 只取第一个Sheet页 // 只取第一个Sheet页
sheet = workbook.getSheetAt(0); sheet = workbook.getSheetAt(0);

9
ddp/pom.xml
View File

@@ -247,13 +247,18 @@
<dependency> <dependency>
<groupId>org.jpmml</groupId> <groupId>org.jpmml</groupId>
<artifactId>pmml-evaluator</artifactId> <artifactId>pmml-evaluator</artifactId>
<version>1.4.1</version> <version>1.5.11</version>
</dependency> </dependency>
<dependency> <dependency>
<groupId>org.jpmml</groupId> <groupId>org.jpmml</groupId>
<artifactId>pmml-evaluator-extension</artifactId> <artifactId>pmml-evaluator-extension</artifactId>
<version>1.4.1</version> <version>1.5.11</version>
</dependency> </dependency>
<!--<dependency>
<groupId>org.jpmml</groupId>
<artifactId>pmml-model</artifactId>
<version>1.7.1</version>
</dependency>-->
<!-- swagger --> <!-- swagger -->
<dependency> <dependency>