84 lines
2.5 KiB
Python
84 lines
2.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
统计计算模块
|
|
"""
|
|
|
|
from typing import List
|
|
import numpy as np
|
|
|
|
from .models import ASRMetrics, AggregatedMetrics
|
|
|
|
|
|
def calculate_percentile(values: List[float], percentile: float) -> float:
|
|
"""
|
|
计算百分位数
|
|
|
|
Args:
|
|
values: 数值列表
|
|
percentile: 百分位 (0-100)
|
|
|
|
Returns:
|
|
百分位值
|
|
"""
|
|
if not values:
|
|
return 0.0
|
|
return float(np.percentile(values, percentile))
|
|
|
|
|
|
def calculate_statistics(
|
|
metrics_list: List[ASRMetrics],
|
|
concurrency_level: int,
|
|
total_test_time: float,
|
|
) -> AggregatedMetrics:
|
|
"""
|
|
计算 ASR 指标统计
|
|
|
|
Args:
|
|
metrics_list: ASR 指标列表
|
|
concurrency_level: 并发级别
|
|
total_test_time: 总测试时间 (秒)
|
|
|
|
Returns:
|
|
聚合后的指标
|
|
"""
|
|
if not metrics_list:
|
|
raise ValueError("指标列表不能为空")
|
|
|
|
successful = [m for m in metrics_list if m.success]
|
|
failed = [m for m in metrics_list if not m.success]
|
|
|
|
# 提取各项指标值
|
|
first_latencies = [
|
|
m.first_result_latency_ms for m in successful if m.first_result_latency_ms is not None
|
|
]
|
|
total_times = [
|
|
m.total_processing_time_ms for m in successful if m.total_processing_time_ms is not None
|
|
]
|
|
rtfs = [m.rtf for m in successful if m.rtf is not None]
|
|
|
|
return AggregatedMetrics(
|
|
concurrency_level=concurrency_level,
|
|
total_requests=len(metrics_list),
|
|
successful_requests=len(successful),
|
|
failed_requests=len(failed),
|
|
total_test_time_seconds=total_test_time,
|
|
# 首次延迟
|
|
first_latency_avg=float(np.mean(first_latencies)) if first_latencies else 0.0,
|
|
first_latency_p50=calculate_percentile(first_latencies, 50),
|
|
first_latency_p95=calculate_percentile(first_latencies, 95),
|
|
first_latency_p99=calculate_percentile(first_latencies, 99),
|
|
first_latency_max=max(first_latencies) if first_latencies else 0.0,
|
|
# 总时间
|
|
total_time_avg=float(np.mean(total_times)) if total_times else 0.0,
|
|
total_time_p50=calculate_percentile(total_times, 50),
|
|
total_time_p95=calculate_percentile(total_times, 95),
|
|
total_time_p99=calculate_percentile(total_times, 99),
|
|
total_time_max=max(total_times) if total_times else 0.0,
|
|
# RTF
|
|
rtf_avg=float(np.mean(rtfs)) if rtfs else 0.0,
|
|
rtf_p50=calculate_percentile(rtfs, 50),
|
|
rtf_p95=calculate_percentile(rtfs, 95),
|
|
rtf_p99=calculate_percentile(rtfs, 99),
|
|
rtf_max=max(rtfs) if rtfs else 0.0,
|
|
)
|