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 TextClassificationParamVo {
    @ApiModelProperty(value = "模型")
    private String modelType;

    @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;

    private Integer computingResourceId;
}
