package com.ruoyi.platform.vo;

import com.fasterxml.jackson.annotation.JsonInclude;
import com.fasterxml.jackson.databind.PropertyNamingStrategy;
import com.fasterxml.jackson.databind.annotation.JsonNaming;
import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import lombok.Data;

import java.util.Map;

@Data
@JsonNaming(PropertyNamingStrategy.SnakeCaseStrategy.class)
@JsonInclude(JsonInclude.Include.NON_NULL)
@ApiModel(description = "视频分类参数")
public class VideoClassificationParamVo {

    @ApiModelProperty(value = "类别数量")
    private Integer numClasses;

    @ApiModelProperty(value = "数据集")
    private Map<String, Object> dataset;

    @ApiModelProperty(value = "epochs")
    private Integer epochs;

    @ApiModelProperty(value = "batch_size")
    private Integer batchSize;

    @ApiModelProperty(value = "学习率")
    private Float lr;

    @ApiModelProperty(value = "是否验证")
    private Boolean isValidate;

    @ApiModelProperty(value = "训练集路径")
    private String trainDataPrefix;

    @ApiModelProperty(value = "验证集路径")
    private String validDataPrefix;

    @ApiModelProperty(value = "训练集标注文件")
    private String trainFilePath;

    @ApiModelProperty(value = "验证集标注文件")
    private String validFilePath;

    private Integer computingResourceId;

}
