自动超参数寻优实验功能开发

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

@@ -23,6 +23,9 @@ public class Ray {
     @ApiModelProperty(value = "数据集")
     private String dataset;
 
+    @ApiModelProperty(value = "数据集挂载路径")
+    private String datasetPath;
+
     @ApiModelProperty(value = "代码")
     private String code;
 

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

@@ -48,10 +48,10 @@ public class RayVo {
     @ApiModelProperty(value = "指标最大化或最小化，min or max")
     private String mode;
 
-    @ApiModelProperty(value = "搜索算法为ASHA，HyperBand时传入，每次试验的最大时间单位。测试将在max_t时间单位后停止。")
+    @ApiModelProperty(value = "单次试验最大时间：单位秒，搜索算法为ASHA，HyperBand时传入，每次试验的最大时间单位。测试将在max_t时间单位后停止。")
     private Integer maxT;
 
-    @ApiModelProperty(value = "搜索算法为MedianStopping时传入，计算中位数的最小试验数。")
+    @ApiModelProperty(value = "计算中位数的最小试验数：搜索算法为MedianStopping时传入，计算中位数的最小试验数。")
     private Integer minSamplesRequired;
 
     @ApiModelProperty(value = "使用cpu数")
@@ -76,4 +76,7 @@ public class RayVo {
     private Map<String, Object> code;
 
     private Map<String, Object> dataset;
+
+    @ApiModelProperty(value = "数据集挂载路径")
+    private String datasetPath;
 }

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

@@ -2,10 +2,10 @@
 <!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
 <mapper namespace="com.ruoyi.platform.mapper.RayDao">
     <insert id="save">
-        insert into ray(name, description, dataset, code, main_py, num_samples, parameters, points_to_evaluate, storage_path,
+        insert into ray(name, description, dataset, dataset_path, code, main_py, num_samples, parameters, points_to_evaluate, storage_path,
                         search_alg, scheduler, metric, mode, max_t,
                         min_samples_required, cpu, gpu, create_by, update_by)
-        values (#{ray.name}, #{ray.dataset}, #{ray.code}, #{ray.mainPy}, #{ray.numSamples}, #{ray.parameters},
+        values (#{ray.name}, #{ray.dataset}, #{ray.datasetPath}, #{ray.code}, #{ray.mainPy}, #{ray.numSamples}, #{ray.parameters},
                 #{ray.pointsToEvaluate}, #{ray.storagePath},
                 #{ray.searchAlg}, #{ray.scheduler}, #{ray.metric}, #{ray.mode}, #{ray.maxT}, #{ray.minSamplesRequired},
                 #{ray.cpu}, #{ray.gpu}, #{ray.createBy}, #{ray.updateBy})
@@ -23,6 +23,9 @@
             <if test="ray.dataset != null and ray.dataset !=''">
                 dataset = #{ray.dataset},
             </if>
+            <if test="ray.datasetPath != null and ray.datasetPath !=''">
+                dataset_path = #{ray.datasetPath},
+            </if>
             <if test="ray.code != null and ray.code !=''">
                 code = #{ray.code},
             </if>