package com.kamco.cd.training.model.dto;

import lombok.AllArgsConstructor;
import lombok.Getter;
import lombok.NoArgsConstructor;
import lombok.Setter;

public class ModelConfigDto {

  @Getter
  @Setter
  @NoArgsConstructor
  @AllArgsConstructor
  public static class Basic {
    private Long configId;
    private Long modelId;
    private Integer epochCount;
    private Float trainPercent;
    private Float validationPercent;
    private Float testPercent;
    private String memo;
  }

  @Getter
  @Setter
  @NoArgsConstructor
  @AllArgsConstructor
  public static class TransferBasic {
    private Long configId;
    private Long modelId;
    private Integer epochCount;
    private Float trainPercent;
    private Float validationPercent;
    private Float testPercent;
    private String memo;
    private Long beforeConfigId;
    private Long beforeModelId;
    private Integer beforeEpochCount;
    private Float beforeTrainPercent;
    private Float beforeValidationPercent;
    private Float beforeTestPercent;
    private String beforeMemo;
  }
}
