package cn.iocoder.yudao.module.llm.controller.admin.training.vo;

import io.swagger.v3.oas.annotations.media.Schema;
import lombok.*;
import java.util.*;
import java.math.BigDecimal;
import org.springframework.format.annotation.DateTimeFormat;
import java.time.LocalDateTime;
import com.alibaba.excel.annotation.*;

@Schema(description = "管理后台 - 训练 Response VO")
@Data
@ExcelIgnoreUnannotated
public class TrainingRespVO {

    @Schema(description = "自增ID", requiredMode = Schema.RequiredMode.REQUIRED, example = "11614")
    @ExcelProperty("自增ID")
    private Integer id;

    @Schema(description = "基座模型名称", requiredMode = Schema.RequiredMode.REQUIRED, example = "张三")
    @ExcelProperty("基座模型名称")
    private String modelName;

    @Schema(description = "介绍", example = "你说的对")
    @ExcelProperty("介绍")
    private String description;

    @Schema(description = "训练轮次", requiredMode = Schema.RequiredMode.REQUIRED)
    @ExcelProperty("训练轮次")
    private Integer trainingEpochs;

    @Schema(description = "启用Lora", requiredMode = Schema.RequiredMode.REQUIRED)
    @ExcelProperty("启用Lora")
    private Boolean useLora;

    @Schema(description = "训练批次", requiredMode = Schema.RequiredMode.REQUIRED)
    @ExcelProperty("训练批次")
    private Integer trainingBatchSize;

    @Schema(description = "评估批次", requiredMode = Schema.RequiredMode.REQUIRED)
    @ExcelProperty("评估批次")
    private Integer evaluationBatchSize;

    @Schema(description = "梯度累加步数", requiredMode = Schema.RequiredMode.REQUIRED)
    @ExcelProperty("梯度累加步数")
    private Integer gradientAccumulationSteps;

    @Schema(description = "使用GPU数量", requiredMode = Schema.RequiredMode.REQUIRED, example = "22492")
    @ExcelProperty("使用GPU数量")
    private Integer gpuCount;

    @Schema(description = "学习率", requiredMode = Schema.RequiredMode.REQUIRED)
    @ExcelProperty("学习率")
    private BigDecimal learningRate;

    @Schema(description = "最大长度", requiredMode = Schema.RequiredMode.REQUIRED)
    @ExcelProperty("最大长度")
    private Integer maxLength;

    @Schema(description = "记录创建时间", requiredMode = Schema.RequiredMode.REQUIRED)
    @ExcelProperty("记录创建时间")
    private LocalDateTime createdAt;

    @Schema(description = "记录更新时间", requiredMode = Schema.RequiredMode.REQUIRED)
    @ExcelProperty("记录更新时间")
    private LocalDateTime updatedAt;

}