Struct TrainingData 

pub struct TrainingData {
    pub system: Option<String>,
    pub prompt: String,
    pub chosen: String,
    pub rejected: Option<String>,
    pub format: TrainingFormat,
    pub metadata: TrainingMetadata,
}

LoRA 学習用データ形式

DPO/SFT 両対応を想定した設計。 初期実装は SFT のみで開始し、データが蓄積されたら DPO に移行予定。

Fields

system: Option<String>

システムプロンプト（オプション）

prompt: String

入力プロンプト（ユーザー入力）

chosen: String

選択された応答（成功ケース）

rejected: Option<String>

拒否された応答（失敗ケース、DPO 用） SFT の場合は None

format: TrainingFormat

学習形式

metadata: TrainingMetadata

メタデータ

Implementations

impl TrainingData

pub fn sft(system: &str, prompt: &str, response: &str) -> Self

SFT 形式のデータを作成（システムプロンプト付き）

Arguments

• system - システムプロンプト
• prompt - ユーザープロンプト
• response - アシスタントレスポンス（chosen）

pub fn sft_simple(prompt: &str, response: &str) -> Self

SFT 形式のデータを作成（シンプル）

Arguments

• prompt - プロンプト
• response - レスポンス（chosen）

pub fn dpo(prompt: &str, chosen: &str, rejected: &str) -> Self

DPO 形式のデータを作成

Arguments

• prompt - プロンプト
• chosen - 選択された応答（成功ケース）
• rejected - 拒否された応答（失敗ケース）

pub fn dpo_with_system( system: &str, prompt: &str, chosen: &str, rejected: &str, ) -> Self

DPO 形式のデータを作成（システムプロンプト付き）

pub fn with_episode_id(self, episode_id: String) -> Self

Episode ID を設定

pub fn with_outcome_score(self, score: f64) -> Self

Outcome スコアを設定

pub fn with_model(self, model: &str) -> Self

モデル名を設定

pub fn with_lora(self, lora: Option<String>) -> Self

LoRA 名を設定

pub fn with_strategy(self, strategy: &str) -> Self

Strategy 名を設定

pub fn with_scenario(self, scenario: &str) -> Self

シナリオ名を設定

pub fn with_custom( self, key: impl Into<String>, value: impl Into<String>, ) -> Self

カスタムメタデータを追加

pub fn is_sft(&self) -> bool

SFT 形式かどうか

pub fn is_dpo(&self) -> bool

DPO 形式かどうか

pub fn is_valid(&self) -> bool

有効なデータかどうか（prompt と chosen が非空）

pub fn is_valid_dpo(&self) -> bool

DPO として有効かどうか（rejected も必要）

impl TrainingData

pub fn to_conversation(&self) -> ConversationData

Conversation 形式に変換

Trait Implementations

impl Clone for TrainingData

fn clone(&self) -> TrainingData

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for TrainingData

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for TrainingData

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl From<&TrainingData> for ConversationData

fn from(data: &TrainingData) -> Self

Converts to this type from the input type.

impl Serialize for TrainingData

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

where __S: Serializer,

Serialize this value into the given Serde serializer.