Skip to main content

mistral_openapi_client/models/
completion_training_parameters.rs

1/*
2 * Mistral AI API
3 *
4 * Our Chat Completion and Embeddings APIs specification. Create your account on [La Plateforme](https://console.mistral.ai) to get access and read the [docs](https://docs.mistral.ai) to learn how to use it.
5 *
6 * The version of the OpenAPI document: 1.0.0
7 * 
8 * Generated by: https://openapi-generator.tech
9 */
10
11use crate::models;
12use serde::{Deserialize, Serialize};
13
14#[derive(Clone, Default, Debug, PartialEq, Serialize, Deserialize)]
15pub struct CompletionTrainingParameters {
16    #[serde(rename = "training_steps", default, with = "::serde_with::rust::double_option", skip_serializing_if = "Option::is_none")]
17    pub training_steps: Option<Option<i32>>,
18    #[serde(rename = "learning_rate", skip_serializing_if = "Option::is_none")]
19    pub learning_rate: Option<f64>,
20    #[serde(rename = "weight_decay", default, with = "::serde_with::rust::double_option", skip_serializing_if = "Option::is_none")]
21    pub weight_decay: Option<Option<f64>>,
22    #[serde(rename = "warmup_fraction", default, with = "::serde_with::rust::double_option", skip_serializing_if = "Option::is_none")]
23    pub warmup_fraction: Option<Option<f64>>,
24    #[serde(rename = "epochs", default, with = "::serde_with::rust::double_option", skip_serializing_if = "Option::is_none")]
25    pub epochs: Option<Option<f64>>,
26    #[serde(rename = "seq_len", default, with = "::serde_with::rust::double_option", skip_serializing_if = "Option::is_none")]
27    pub seq_len: Option<Option<i32>>,
28    #[serde(rename = "fim_ratio", default, with = "::serde_with::rust::double_option", skip_serializing_if = "Option::is_none")]
29    pub fim_ratio: Option<Option<f64>>,
30}
31
32impl CompletionTrainingParameters {
33    pub fn new() -> CompletionTrainingParameters {
34        CompletionTrainingParameters {
35            training_steps: None,
36            learning_rate: None,
37            weight_decay: None,
38            warmup_fraction: None,
39            epochs: None,
40            seq_len: None,
41            fim_ratio: None,
42        }
43    }
44}
45