diff --git a/src/main/java/com/kamco/cd/kamcoback/inference/service/InferenceResultService.java b/src/main/java/com/kamco/cd/kamcoback/inference/service/InferenceResultService.java index 268694f0..43023bae 100644 --- a/src/main/java/com/kamco/cd/kamcoback/inference/service/InferenceResultService.java +++ b/src/main/java/com/kamco/cd/kamcoback/inference/service/InferenceResultService.java @@ -414,9 +414,9 @@ public class InferenceResultService { String modelType = ""; - if (modelInfo.getModelType().equals(ModelType.M1.getId())) { + if (modelInfo.getModelType().equals(ModelType.G1.getId())) { modelType = "G1"; - } else if (modelInfo.getModelType().equals(ModelType.M2.getId())) { + } else if (modelInfo.getModelType().equals(ModelType.G2.getId())) { modelType = "G2"; } else { modelType = "G3"; diff --git a/src/main/java/com/kamco/cd/kamcoback/model/dto/ModelMngDto.java b/src/main/java/com/kamco/cd/kamcoback/model/dto/ModelMngDto.java index a85387a9..6cf752bc 100644 --- a/src/main/java/com/kamco/cd/kamcoback/model/dto/ModelMngDto.java +++ b/src/main/java/com/kamco/cd/kamcoback/model/dto/ModelMngDto.java @@ -21,9 +21,9 @@ public class ModelMngDto { @Getter @AllArgsConstructor public enum ModelType implements EnumType { - M1("모델 M1"), - M2("모델 M2"), - M3("모델 M3"); + G1("G1"), + G2("G2"), + G3("G3"); private final String desc;