com.theokanning.openai.finetune.HyperParameters Maven / Gradle / Ivy

Go to download
package com.theokanning.openai.finetune;

import com.fasterxml.jackson.annotation.JsonProperty;
import lombok.Data;

/**
 * Fine-tuning job hyperparameters
 *
 * https://beta.openai.com/docs/api-reference/fine-tunes
 */
@Deprecated
@Data
public class HyperParameters {

    /**
     * The batch size to use for training.
     */
    @JsonProperty("batch_size")
    Integer batchSize;

    /**
     * The learning rate multiplier to use for training.
     */
    @JsonProperty("learning_rate_multiplier")
    Double learningRateMultiplier;

    /**
     * The number of epochs to train the model for.
     */
    @JsonProperty("n_epochs")
    Integer nEpochs;

    /**
     * The weight to use for loss on the prompt tokens.
     */
    @JsonProperty("prompt_loss_weight")
    Double promptLossWeight;
}