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 java.io.IOException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.*;

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

// 调argo转换接口
try {
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 Float lr;

private String experimentInsId;
}

Loading…
Cancel
Save