Browse Source

优化主动学习

pull/225/head
chenzhihang 9 months ago
parent
commit
738b49fd46
2 changed files with 5 additions and 5 deletions
  1. +3
    -5
      ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ActiveLearnServiceImpl.java
  2. +2
    -0
      ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/vo/ActiveLearnParamVo.java

+ 3
- 5
ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ActiveLearnServiceImpl.java View File

@@ -26,10 +26,7 @@ import org.springframework.transaction.annotation.Transactional;


import javax.annotation.Resource; import javax.annotation.Resource;
import java.io.IOException; import java.io.IOException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.*;


@Service @Service
public class ActiveLearnServiceImpl implements ActiveLearnService { public class ActiveLearnServiceImpl implements ActiveLearnService {
@@ -150,8 +147,9 @@ public class ActiveLearnServiceImpl implements ActiveLearnService {
activeLearnParamVo.setDataset(JsonUtils.jsonToMap(activeLearn.getDataset())); activeLearnParamVo.setDataset(JsonUtils.jsonToMap(activeLearn.getDataset()));
activeLearnParamVo.setModel(JsonUtils.jsonToMap(activeLearn.getModel())); activeLearnParamVo.setModel(JsonUtils.jsonToMap(activeLearn.getModel()));
activeLearnParamVo.setImage(JsonUtils.jsonToMap(activeLearn.getImage())); activeLearnParamVo.setImage(JsonUtils.jsonToMap(activeLearn.getImage()));
activeLearnParamVo.setExperimentInsId(String.valueOf(UUID.randomUUID()));
String param = JsonUtils.getConvertParam(activeLearnParamVo); String param = JsonUtils.getConvertParam(activeLearnParamVo);

// 调argo转换接口 // 调argo转换接口
try { try {
String convertRes = HttpUtils.sendPost(argoUrl + convertActiveLearn, param); String convertRes = HttpUtils.sendPost(argoUrl + convertActiveLearn, param);


+ 2
- 0
ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/vo/ActiveLearnParamVo.java View File

@@ -64,4 +64,6 @@ public class ActiveLearnParamVo {
private Integer epochs; private Integer epochs;


private Float lr; private Float lr;

private String experimentInsId;
} }

Loading…
Cancel
Save