position_weight.py

"""
Модуль вычисления позиционного веса
"""
import math
import numpy as np


def compute_position_weight(N: int, delta: float = 0.1) -> np.ndarray:
    """
    Вычисление позиционного веса π(i).

    Формула: π(i) = 1 + δ * sqrt(|i - N/2| / (N/2))

    Args:
        N: количество предложений
        delta: сила позиционного эффекта (0-0.2)

    Returns:
        массив позиционных весов
    """
    weights = np.zeros(N)
    for i in range(N):
        pos = i + 1
        weights[i] = 1 + delta * math.sqrt(abs(pos - N / 2) / (N / 2))
    return weights