MVPTeam/test
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

모델관리 수정

Browse Source
This commit is contained in:
Moon
2026-01-09 12:03:42 +09:00
parent 13c8e7b09b
commit 85c7c53660
9 changed files with 195 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
src/main/java/com/kamco/cd/kamcoback/model/dto/ModelMngDto.java
View File

@@ -59,6 +59,12 @@ public class ModelMngDto {
private String modelType;
private String filePath;
private String fileName;
private String cdModelPath;
private String cdModelFileName;
private String cdModelConfig;
private String cdModelConfigFileName;
private String clsModelPath;
private String clsModelFileName;
private String memo;
public Basic(
@@ -74,6 +80,12 @@ public class ModelMngDto {
String modelType,
String filePath,
String fileName,
String cdModelPath,
String cdModelFileName,
String cdModelConfig,
String cdModelConfigFileName,
String clsModelPath,
String clsModelFileName,
String memo) {
this.modelUid = modelUid;
this.modelVer = modelVer;
@@ -87,6 +99,12 @@ public class ModelMngDto {
this.modelType = modelType;
this.filePath = filePath;
this.fileName = fileName;
this.cdModelPath = cdModelPath;
this.cdModelFileName = cdModelFileName;
this.cdModelConfig = cdModelConfig;
this.cdModelConfigFileName = cdModelConfigFileName;
this.clsModelPath = clsModelPath;
this.clsModelFileName = clsModelFileName;
this.memo = memo;
}
}
@@ -123,6 +141,12 @@ public class ModelMngDto {
private String modelVer;
private String filePath;
private String fileName;
private String cdModelPath;
private String cdModelFileName;
private String cdModelConfigPath;
private String cdModelConfigFileName;
private String clsModelPath;
private String clsModelFileName;
private String memo;
@JsonIgnore private UUID uuid;
@@ -151,4 +175,21 @@ public class ModelMngDto {
return PageRequest.of(page, size);
}
}
@Schema(name = "ModelMetricAddReq", description = "모델 등록 req")
@Getter
@Setter
@NoArgsConstructor
@AllArgsConstructor
public static class ModelMetricAddReq {
private Long modelUid;
private Long modelVerUid;
private double f1Score;
private double precision;
private double recall;
private double loss;
private double iou;
}
}

15
src/main/java/com/kamco/cd/kamcoback/model/service/ModelMngService.java
View File

@@ -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, "등록되었습니다.");
}