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) -> TrainingData

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

Arguments

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

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

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

Arguments

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

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

DPO 形式のデータを作成

Arguments

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

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

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

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

Episode ID を設定

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

Outcome スコアを設定

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

モデル名を設定

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

LoRA 名を設定

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

Strategy 名を設定

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

シナリオ名を設定

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

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

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<(), Error>

Formats the value using the given formatter.

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

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

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl From<&TrainingData> for ConversationData

fn from(data: &TrainingData) -> ConversationData

Converts to this type from the input type.

impl Serialize for TrainingData

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

where __S: Serializer,

Serialize this value into the given Serde serializer.