mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2026-05-26 04:57:52 +08:00
Delete now empty file
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com> Signed-off-by: weichen <calvin_zhu0210@outlook.com>
This commit is contained in:
parent
58615e5889
commit
da9d153112
@ -1,112 +0,0 @@
|
|||||||
# SPDX-License-Identifier: Apache-2.0
|
|
||||||
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
||||||
import math
|
|
||||||
|
|
||||||
from vllm.logger import init_logger
|
|
||||||
|
|
||||||
logger = init_logger(__name__)
|
|
||||||
|
|
||||||
|
|
||||||
class ScoreDim:
|
|
||||||
"""
|
|
||||||
Normalized scoring dimension.
|
|
||||||
"""
|
|
||||||
|
|
||||||
def __init__(
|
|
||||||
self, name: str, median: float, norm_scale=0.0, weight=0.5, reverse=False
|
|
||||||
):
|
|
||||||
self.name = name
|
|
||||||
self.median = median
|
|
||||||
if norm_scale != 0.0:
|
|
||||||
self.norm_scale = norm_scale
|
|
||||||
else:
|
|
||||||
self.norm_scale = 1 / median
|
|
||||||
self.weight = weight
|
|
||||||
self.reverse = reverse
|
|
||||||
|
|
||||||
|
|
||||||
class NormalizedScorer:
|
|
||||||
"""
|
|
||||||
Normalize unbounded N-dimensional values into a composite score using the Sigmoid
|
|
||||||
function.
|
|
||||||
"""
|
|
||||||
|
|
||||||
def __init__(self, dim_list: list[ScoreDim]) -> None:
|
|
||||||
"""
|
|
||||||
Initialize the scorer with a list of scoring dimensions.
|
|
||||||
|
|
||||||
Args:
|
|
||||||
dim_list: A list of `ScoreDim` objects. Each dimension must define a
|
|
||||||
median reference point, scaling factor, and weight.
|
|
||||||
"""
|
|
||||||
self.dim_list = dim_list
|
|
||||||
self.dim_count = len(dim_list)
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def _sigmoid_normalize(value, median, norm_scale):
|
|
||||||
"""Sigmoid function: Maps value to (0, 1)."""
|
|
||||||
return 1 / (1 + math.exp(-norm_scale * (value - median)))
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def _inv_sigmoid_normalize(value, median, norm_scale):
|
|
||||||
"""Inverse Sigmoid: Used for dimensions where a larger value yields a lower
|
|
||||||
score.
|
|
||||||
"""
|
|
||||||
# Equivalent to sigmoid(-x), but more numerically stable.
|
|
||||||
return 1 / (1 + math.exp(norm_scale * (value - median)))
|
|
||||||
|
|
||||||
def score(self, *dims: float) -> float:
|
|
||||||
"""
|
|
||||||
Compute the composite score.
|
|
||||||
Larger value → higher score → use forward Sigmoid.
|
|
||||||
Smaller value → higher score → use inverse Sigmoid.
|
|
||||||
"""
|
|
||||||
if len(dims) > self.dim_count:
|
|
||||||
raise ValueError(
|
|
||||||
f"Dim num({len(dims)}) exceeds max num dim({self.dim_count})"
|
|
||||||
)
|
|
||||||
|
|
||||||
final_score = 0.0
|
|
||||||
for idx, dim_value in enumerate(dims):
|
|
||||||
dim_info = self.dim_list[idx]
|
|
||||||
if dim_info.reverse:
|
|
||||||
score = self._inv_sigmoid_normalize(
|
|
||||||
dim_value, dim_info.median, dim_info.norm_scale
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
score = self._sigmoid_normalize(
|
|
||||||
dim_value, dim_info.median, dim_info.norm_scale
|
|
||||||
)
|
|
||||||
logger.debug("%s(%s) : %.10f", dim_info.name, dim_info.reverse, score)
|
|
||||||
|
|
||||||
# Weighted summation.
|
|
||||||
final_score += score * dim_info.weight
|
|
||||||
return max(0.0, min(1.0, final_score)) # Clamp to [0, 1].
|
|
||||||
|
|
||||||
|
|
||||||
class TimeAndLengthScorer(NormalizedScorer):
|
|
||||||
"""
|
|
||||||
Scorer for time and length dimensions; defaults to forward scoring with equal
|
|
||||||
weights (0.5 each).
|
|
||||||
"""
|
|
||||||
|
|
||||||
def __init__(
|
|
||||||
self,
|
|
||||||
time_median,
|
|
||||||
length_median,
|
|
||||||
time_scale=0.0,
|
|
||||||
length_scale=0.0,
|
|
||||||
time_weight=0.5,
|
|
||||||
length_weight=0.5,
|
|
||||||
reverse_time=False,
|
|
||||||
reverse_len=False,
|
|
||||||
) -> None:
|
|
||||||
dim_list = [
|
|
||||||
ScoreDim("time", time_median, time_scale, time_weight, reverse_time),
|
|
||||||
ScoreDim("length", length_median, length_scale, length_weight, reverse_len),
|
|
||||||
]
|
|
||||||
super().__init__(dim_list)
|
|
||||||
|
|
||||||
def score(self, *dims: float) -> float:
|
|
||||||
assert len(dims) == 2
|
|
||||||
return super().score(*dims)
|
|
||||||
Loading…
x
Reference in New Issue
Block a user