iterative_selection.py

"""
Модуль итеративного отбора предложений с подавлением семантической избыточности
"""
import numpy as np


def iterative_selection(scores: np.ndarray, similarity_matrix: np.ndarray,
                        k: int, theta: float = 0.6) -> list:
    """
    Итеративный жадный отбор с подавлением семантически близких предложений.

    Args:
        scores: начальные оценки Score(S_i)
        similarity_matrix: матрица попарного семантического сходства
        k: целевое количество предложений
        theta: коэффициент подавления (0-1)

    Returns:
        индексы отобранных предложений в хронологическом порядке
    """
    N = len(scores)
    selected = []
    remaining = list(range(N))
    current_scores = scores.copy()

    for _ in range(min(k, N)):
        best_idx = max(remaining, key=lambda i: current_scores[i])
        selected.append(best_idx)
        remaining.remove(best_idx)

        for i in remaining:
            sim = similarity_matrix[best_idx, i]
            current_scores[i] = current_scores[i] * (1 - theta * sim)
            current_scores[i] = max(current_scores[i], 0.0)

    return sorted(selected)