diff --git a/src/main/java/com/kamco/cd/training/postgres/entity/ModelMasterEntity.java b/src/main/java/com/kamco/cd/training/postgres/entity/ModelMasterEntity.java
index a06380e..05311dd 100644
--- a/src/main/java/com/kamco/cd/training/postgres/entity/ModelMasterEntity.java
+++ b/src/main/java/com/kamco/cd/training/postgres/entity/ModelMasterEntity.java
@@ -91,6 +91,12 @@ public class ModelMasterEntity {
   @Column(name = "before_model_id")
   private Long beforeModelId;
 
+  @Column(name = "step1_metric_save_yn")
+  private Boolean step1MetricSaveYn;
+
+  @Column(name = "step2_metric_save_yn")
+  private Boolean step2MetricSaveYn;
+
   public ModelTrainMngDto.Basic toDto() {
     return new ModelTrainMngDto.Basic(
         this.id,