Struct CreateFineTuneParameters

Source
pub struct CreateFineTuneParameters {
    pub training_file: String,
    pub validation_file: Option<String>,
    pub model: Option<String>,
    pub epochs: Option<u32>,
    pub batch_size: Option<u32>,
    pub learning_rate_multiplier: Option<f32>,
    pub prompt_loss_weight: Option<f32>,
    pub compute_classification_metrics: Option<bool>,
    pub classification_n_classes: Option<u32>,
    pub classification_positive_class: Option<String>,
    pub classification_beta: Option<f32>,
    pub suffix: Option<String>,
}

Fields§

§training_file: String

The ID of an uploaded file that contains training data.

§validation_file: Option<String>

The ID of an uploaded file that contains validation data.

§model: Option<String>

The name of the base model to use for fine-tuning.

§epochs: Option<u32>

The number of epochs to train the model for.

§batch_size: Option<u32>

The batch size to use for training.

§learning_rate_multiplier: Option<f32>

The learning rate multiplier to use for training. The fine-tuning learning rate is the original learning rate used for pretraining multiplied by this value.

§prompt_loss_weight: Option<f32>

The weight to use for loss on the prompt tokens. This controls how much the model tries to learn to generate the prompt (as compared to the completion which always has a weight of 1.0), and can add a stabilizing effect to training when completions are short.

§compute_classification_metrics: Option<bool>

If set, we calculate classification-specific metrics such as accuracy and F-1 score using the validation set at the end of every epoch. These metrics can be viewed in the results file.

In order to compute classification metrics, you must provide a validation_file. Additionally, you must specify classification_n_classes for multiclass classification or classification_positive_class for binary classification.

§classification_n_classes: Option<u32>

The number of classes in a classification task.

This parameter is required for multiclass classification.

§classification_positive_class: Option<String>

The positive class in binary classification.

This parameter is needed to generate precision, recall, and F1 metrics when doing binary classification.

§classification_beta: Option<f32>

If this is provided, we calculate F-beta scores at the specified beta values. The F-beta score is a generalization of F-1 score. This is only used for binary classification.

With a beta of 1 (i.e. the F-1 score), precision and recall are given the same weight. A larger beta score puts more weight on recall and less on precision. A smaller beta score puts more weight on precision and less on recall.

§suffix: Option<String>

A string of up to 40 characters that will be added to your fine-tuned model name.

For example, a suffix of “custom-model-name” would produce a model name like ada:ft-your-org:custom-model-name-2022-02-15-04-21-04.

Implementations§

Source§

impl CreateFineTuneParameters

Source

pub fn new(training_file: String) -> Self

Source

pub fn validation_file(self, validation_file: String) -> Self

Source

pub fn model(self, model: String) -> Self

Source

pub fn epochs(self, epochs: u32) -> Self

Source

pub fn batch_size(self, batch_size: u32) -> Self

Source

pub fn learning_rate_multiplier(self, learning_rate_multiplier: f32) -> Self

Source

pub fn prompt_loss_weight(self, prompt_loss_weight: f32) -> Self

Source

pub fn compute_classification_metrics( self, compute_classification_metrics: bool, ) -> Self

Source

pub fn classification_n_classes(self, classification_n_classes: u32) -> Self

Source

pub fn classification_positive_class( self, classification_positive_class: String, ) -> Self

Source

pub fn classification_beta(self, classification_beta: f32) -> Self

Source

pub fn suffix(self, suffix: String) -> Self

Source

pub fn build(self) -> Result<CreateFineTuneParameters, String>

Trait Implementations§

Source§

impl Debug for CreateFineTuneParameters

Source§

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter. Read more
Source§

impl<'de> Deserialize<'de> for CreateFineTuneParameters

Source§

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>
where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer. Read more
Source§

impl Serialize for CreateFineTuneParameters

Source§

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>
where __S: Serializer,

Serialize this value into the given Serde serializer. Read more

Auto Trait Implementations§

Blanket Implementations§

Source§

impl<T> Any for T
where T: 'static + ?Sized,

Source§

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more
Source§

impl<T> Borrow<T> for T
where T: ?Sized,

Source§

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more
Source§

impl<T> BorrowMut<T> for T
where T: ?Sized,

Source§

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more
Source§

impl<T> From<T> for T

Source§

fn from(t: T) -> T

Returns the argument unchanged.

Source§

impl<T> Instrument for T

Source§

fn instrument(self, span: Span) -> Instrumented<Self>

Instruments this type with the provided Span, returning an Instrumented wrapper. Read more
Source§

fn in_current_span(self) -> Instrumented<Self>

Instruments this type with the current Span, returning an Instrumented wrapper. Read more
Source§

impl<T, U> Into<U> for T
where U: From<T>,

Source§

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

Source§

impl<T> Same for T

Source§

type Output = T

Should always be Self
Source§

impl<T, U> TryFrom<U> for T
where U: Into<T>,

Source§

type Error = Infallible

The type returned in the event of a conversion error.
Source§

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

Performs the conversion.
Source§

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

Source§

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.
Source§

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.
Source§

impl<V, T> VZip<V> for T
where V: MultiLane<T>,

Source§

fn vzip(self) -> V

Source§

impl<T> WithSubscriber for T

Source§

fn with_subscriber<S>(self, subscriber: S) -> WithDispatch<Self>
where S: Into<Dispatch>,

Attaches the provided Subscriber to this type, returning a WithDispatch wrapper. Read more
Source§

fn with_current_subscriber(self) -> WithDispatch<Self>

Attaches the current default Subscriber to this type, returning a WithDispatch wrapper. Read more
Source§

impl<T> DeserializeOwned for T
where T: for<'de> Deserialize<'de>,