DATA_MIDDLE_PLATFORM/Data_Middle_Platform_Server
3
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Merge remote-tracking branch 'origin/master'

Browse Source
This commit is contained in:
limin 2025-01-16 20:36:23 +08:00
commit dc408213d4
1 changed files with 33 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/modelassesstaskauto/ModelAssessTaskAutoInfoController.java
View File

@ -22,7 +22,10 @@ import javax.annotation.Resource;
import javax.servlet.http.HttpServletResponse;
import javax.validation.Valid;
import java.io.IOException;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
import static cn.iocoder.yudao.framework.apilog.core.enums.OperateTypeEnum.EXPORT;
import static cn.iocoder.yudao.framework.common.pojo.CommonResult.success;
@ -69,6 +72,36 @@ public class ModelAssessTaskAutoInfoController {
return success(BeanUtils.toBean(modelAssessTaskAutoInfo, ModelAssessTaskAutoInfoRespVO.class));
}
@GetMapping("/get/count")
@Operation(summary = "获得雷达图数据")
public CommonResult<Map<Integer,Double>> count(@RequestParam("taskId") Long taskId) {
ModelAssessTaskAutoInfoPageReqVO modelAssessTaskAutoInfoPageReqVO = new ModelAssessTaskAutoInfoPageReqVO();
modelAssessTaskAutoInfoPageReqVO.setTaskId(taskId);
List<ModelAssessTaskAutoInfoDO> list = modelAssessTaskAutoInfoService.getListByTaskId(modelAssessTaskAutoInfoPageReqVO);
List<Integer> res = list.stream()
.map(ModelAssessTaskAutoInfoDO::getEmotionTendency)
.collect(Collectors.toList());
int size = list.size();
double zero = 0;
double one = 0;
double two = 0;
for (Integer re : res) {
if(re == 0){
zero++;
}else if(re == 1){
one++;
}else if(re == 2){
two++;
}
}
Map<Integer,Double> map = new HashMap<>();
map.put(0,zero * 100 / size);
map.put(1,one * 100 / size);
map.put(2,two * 100 / size);
return success(map);
}
@GetMapping("/list")
@Operation(summary = "根绝任务id获得模型评估任务-自动评估")
@PreAuthorize("@ss.hasPermission('llm:model-assess-task-auto-info:query')")