スコアリングシステム仕様書

pub trait Scorer<Q, C>: Send + Sync {
    /// 実行可否の判定
    fn enable(&self, _query: &Q) -> bool { true }

    /// 候補のスコアリング（非同期）
    async fn score(&self, query: &Q, candidates: &[C]) -> Result<Vec<C>, String>;

    /// 単一候補の更新
    fn update(&self, candidate: &mut C, scored: C);

    /// 全候補の一括更新
    fn update_all(&self, candidates: &mut [C], scored: Vec<C>);
}

let tweet_id = c.retweeted_tweet_id.unwrap_or(c.tweet_id as u64);
let author_id = c.retweeted_user_id.unwrap_or(c.author_id);

let action_probs: HashMap<usize, f64> = distribution
    .top_log_probs
    .iter()
    .enumerate()
    .map(|(idx, log_prob)| (idx, (*log_prob as f64).exp()))
    .collect();

pub struct PhoenixScores {
    pub favorite_score: Option<f64>,        // いいね確率
    pub reply_score: Option<f64>,           // リプライ確率
    pub retweet_score: Option<f64>,         // リツイート確率
    pub photo_expand_score: Option<f64>,    // 画像展開確率
    pub click_score: Option<f64>,           // クリック確率
    pub profile_click_score: Option<f64>,   // プロフィールクリック確率
    pub vqv_score: Option<f64>,             // ビデオ品質視聴確率
    pub share_score: Option<f64>,           // シェア確率
    pub share_via_dm_score: Option<f64>,    // DMシェア確率
    pub share_via_copy_link_score: Option<f64>, // リンクコピー確率
    pub dwell_score: Option<f64>,           // 滞在確率
    pub quote_score: Option<f64>,           // 引用確率
    pub quoted_click_score: Option<f64>,    // 引用クリック確率
    pub follow_author_score: Option<f64>,   // フォロー確率
    pub not_interested_score: Option<f64>,  // 興味なし確率
    pub block_author_score: Option<f64>,    // ブロック確率
    pub mute_author_score: Option<f64>,     // ミュート確率
    pub report_score: Option<f64>,          // 報告確率
    pub dwell_time: Option<f64>,            // 予測滞在時間
}

combined_score =
    favorite_score × FAVORITE_WEIGHT
  + reply_score × REPLY_WEIGHT
  + retweet_score × RETWEET_WEIGHT
  + photo_expand_score × PHOTO_EXPAND_WEIGHT
  + click_score × CLICK_WEIGHT
  + profile_click_score × PROFILE_CLICK_WEIGHT
  + vqv_score × VQV_WEIGHT (条件付き)
  + share_score × SHARE_WEIGHT
  + share_via_dm_score × SHARE_VIA_DM_WEIGHT
  + share_via_copy_link_score × SHARE_VIA_COPY_LINK_WEIGHT
  + dwell_score × DWELL_WEIGHT
  + quote_score × QUOTE_WEIGHT
  + quoted_click_score × QUOTED_CLICK_WEIGHT
  + dwell_time × CONT_DWELL_TIME_WEIGHT
  + follow_author_score × FOLLOW_AUTHOR_WEIGHT
  + not_interested_score × NOT_INTERESTED_WEIGHT
  + block_author_score × BLOCK_AUTHOR_WEIGHT
  + mute_author_score × MUTE_AUTHOR_WEIGHT
  + report_score × REPORT_WEIGHT

fn vqv_weight_eligibility(candidate: &PostCandidate) -> f64 {
    if candidate.video_duration_ms.is_some_and(|ms| ms > MIN_VIDEO_DURATION_MS) {
        VQV_WEIGHT
    } else {
        0.0
    }
}

fn offset_score(combined_score: f64) -> f64 {
    if WEIGHTS_SUM == 0.0 {
        combined_score.max(0.0)
    } else if combined_score < 0.0 {
        (combined_score + NEGATIVE_WEIGHTS_SUM) / WEIGHTS_SUM * NEGATIVE_SCORES_OFFSET
    } else {
        combined_score + NEGATIVE_SCORES_OFFSET
    }
}

impl Default for AuthorDiversityScorer {
    fn default() -> Self {
        Self::new(AUTHOR_DIVERSITY_DECAY, AUTHOR_DIVERSITY_FLOOR)
    }
}

fn multiplier(&self, position: usize) -> f64 {
    (1.0 - self.floor) * self.decay_factor.powf(position as f64) + self.floor
}

match c.in_network {
    Some(false) => base_score * OON_WEIGHT_FACTOR,
    _ => base_score,
}

pub struct PostCandidate {
    // ...
    pub phoenix_scores: PhoenixScores,    // Phoenixモデルの予測結果
    pub weighted_score: Option<f64>,       // 重み付きスコア
    pub score: Option<f64>,                // 最終スコア
    // ...
}