文本分类功能开发

ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/domain/TextClassification.java

@@ -21,7 +21,7 @@ public class TextClassification {
     private String description;
 
     @ApiModelProperty(value = "模型")
-    private String model;
+    private String modelType;
 
     @ApiModelProperty(value = "数据集")
     private String dataset;

ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/TextClassificationServiceImpl.java

@@ -177,7 +177,7 @@ public class TextClassificationServiceImpl implements TextClassificationService
                 List output1 = (ArrayList) param_output.values().toArray()[0];
                 Map<String, String> output2 = (Map<String, String>) output1.get(0);
                 String outputPath = minioEndpoint + "/" + output2.get("path").replace("{{workflow.name}}", (String) metadata.get("name")) + "/";
-                textClassificationIns.setModelPath(outputPath + "/saved_dict/" + textClassification.getModel() + ".ckpt");
+                textClassificationIns.setModelPath(outputPath + "/saved_dict/" + textClassification.getModelType() + ".ckpt");
                 textClassificationInsDao.insert(textClassificationIns);
                 textClassificationInsService.updateTextClassificationStatus(id);
                 // 记录开始扣除积分

ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/vo/TextClassificationParamVo.java

@@ -15,7 +15,7 @@ import java.util.Map;
 @ApiModel(description = "文本分类参数")
 public class TextClassificationParamVo {
     @ApiModelProperty(value = "模型")
-    private String model;
+    private String modelType;
 
     @ApiModelProperty(value = "数据集")
     private Map<String, Object> dataset;

ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/vo/TextClassificationVo.java

@@ -22,7 +22,7 @@ public class TextClassificationVo {
     private String description;
 
     @ApiModelProperty(value = "模型")
-    private String model;
+    private String modelType;
 
     @ApiModelProperty(value = "数据集")
     private Map<String, Object> dataset;

ruoyi-modules/management-platform/src/main/resources/mapper/managementPlatform/TextClassificationDaoMapper.xml

@@ -26,8 +26,8 @@
     </select>
 
     <insert id="save" keyProperty="id" useGeneratedKeys="true">
-        insert into text_classification(name, description, model, dataset, epochs, batch_size, lr, create_by, update_by, computing_resource_id)
-        values (#{textClassification.name}, #{textClassification.description}, #{textClassification.model},
+        insert into text_classification(name, description, model_type, dataset, epochs, batch_size, lr, create_by, update_by, computing_resource_id)
+        values (#{textClassification.name}, #{textClassification.description}, #{textClassification.modelType},
                 #{textClassification.dataset}, #{textClassification.epochs}, #{textClassification.batchSize},
                 #{textClassification.lr}, #{textClassification.createBy}, #{textClassification.updateBy}, #{textClassification.computingResourceId})
     </insert>
@@ -41,8 +41,8 @@
             <if test="textClassification.description != null and textClassification.description !=''">
                 description = #{textClassification.description},
             </if>
-            <if test="textClassification.model != null and textClassification.model !=''">
-                model = #{textClassification.model},
+            <if test="textClassification.modelType != null and textClassification.modelType !=''">
+                model_type = #{textClassification.modelType},
             </if>
             <if test="textClassification.dataset != null and textClassification.dataset !=''">
                 dataset = #{textClassification.dataset},