com.launchableinc.openai.finetune.HyperParameters Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of api Show documentation
Show all versions of api Show documentation
Basic java objects for the OpenAI GPT APIs
The newest version!
package com.launchableinc.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;
}