모델관리 수정
This commit is contained in:
@@ -2,6 +2,7 @@ package com.kamco.cd.kamcoback.model.service;
|
||||
|
||||
import com.kamco.cd.kamcoback.config.api.ApiResponseDto;
|
||||
import com.kamco.cd.kamcoback.model.dto.ModelMngDto;
|
||||
import com.kamco.cd.kamcoback.model.dto.ModelMngDto.ModelMetricAddReq;
|
||||
import com.kamco.cd.kamcoback.postgres.core.ModelMngCoreService;
|
||||
import com.kamco.cd.kamcoback.upload.dto.UploadDto;
|
||||
import com.kamco.cd.kamcoback.upload.service.UploadService;
|
||||
@@ -62,7 +63,19 @@ public class ModelMngService {
|
||||
public ApiResponseDto.ResponseObj insertModel(ModelMngDto.AddReq addReq) {
|
||||
UUID uuid = UUID.randomUUID();
|
||||
addReq.setUuid(uuid);
|
||||
modelMngCoreService.insertModel(addReq);
|
||||
Long modelUid = modelMngCoreService.insertModel(addReq);
|
||||
|
||||
ModelMetricAddReq modelMetricAddReq = new ModelMetricAddReq();
|
||||
modelMetricAddReq.setModelUid(modelUid);
|
||||
modelMetricAddReq.setModelVerUid(modelUid);
|
||||
modelMetricAddReq.setF1Score(0);
|
||||
modelMetricAddReq.setPrecision(0);
|
||||
modelMetricAddReq.setRecall(0);
|
||||
modelMetricAddReq.setLoss(0);
|
||||
modelMetricAddReq.setIou(0);
|
||||
|
||||
modelMngCoreService.insertModelResultMetric(modelMetricAddReq);
|
||||
|
||||
return new ApiResponseDto.ResponseObj(ApiResponseDto.ApiResponseCode.OK, "등록되었습니다.");
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user