Skip to main content

calculate_choice_energy

Function calculate_choice_energy 

Source
pub fn calculate_choice_energy(
    question: &Question,
    logits: &[f64],
    calib_config: &CalibrationConfig,
) -> Option<f64>
Expand description

Choice 型の質問とロジット配列から、正規化ヘルムホルツ自由エネルギーを算出する。

モデル較正設定の energy_temperature (または候補数バケット温度 $T = \text{calib_config.get_temperature(QuestionType::Choice, K)}$) を用い、 $E_{\text{norm}}(x) = -T \ln \sum_{i=1}^K \exp(z_i / T) + T \ln K$ を算出する。 Choice 型以外 (Score 型、Noul 型) またはロジットが空の場合は None を返す。

§引数

  • question: 質問定義。
  • logits: 決定ロジットスライス。
  • calib_config: キャリブレーション設定。

§戻り値

正規化ヘルムホルツ自由エネルギー値 (任意)。