패키징 zip파일 만들기 커밋

src/main/java/com/kamco/cd/training/train/dto/ModelTrainMetricsDto.java

@@ -1,6 +1,8 @@
 package com.kamco.cd.training.train.dto;
 
+import com.fasterxml.jackson.annotation.JsonProperty;
 import io.swagger.v3.oas.annotations.media.Schema;
+import java.util.Properties;
 import java.util.UUID;
 import lombok.AllArgsConstructor;
 import lombok.Getter;
@@ -20,4 +22,38 @@ public class ModelTrainMetricsDto {
     private String responsePath;
     private UUID uuid;
   }
+
+  @Getter
+  @AllArgsConstructor
+  public static class ModelMetricJsonDto {
+
+    @JsonProperty("cd_model_type")
+    private String cdModelType;
+
+    @JsonProperty("model_version")
+    private String modelVersion;
+
+    private Properties properties;
+  }
+
+  @Getter
+  @AllArgsConstructor
+  public static class Properties {
+
+    @JsonProperty("f1_score")
+    private Float f1Score;
+
+    private Float precision;
+    private Float recall;
+    private Float loss;
+    private Double iou;
+  }
+
+  @Getter
+  @AllArgsConstructor
+  public static class ModelTestFileName {
+
+    private String bestEpochFileName;
+    private String modelVersion;
+  }
 }