fix: use multi-model peak window for gaussian probability

This commit is contained in:
2569718930@qq.com
2026-05-28 10:03:29 +08:00
parent fb4a9d7e09
commit be549daf50
4 changed files with 148 additions and 18 deletions
+93 -8
View File
@@ -48,6 +48,91 @@ def _sf(v):
return None
def _median(values: List[float]) -> Optional[float]:
if not values:
return None
sorted_values = sorted(values)
mid = len(sorted_values) // 2
if len(sorted_values) % 2:
return sorted_values[mid]
return (sorted_values[mid - 1] + sorted_values[mid]) / 2.0
def _peak_hours_from_hourly_values(
hourly_values: List[Tuple[str, float]],
*,
tolerance: float = 0.3,
) -> List[str]:
if not hourly_values:
return []
peak_value = max(value for _, value in hourly_values)
return [
time_part
for time_part, value in hourly_values
if abs(value - peak_value) <= tolerance
]
def _resolve_peak_hours(
weather_data: dict,
local_date_str: str,
open_meteo_times: Optional[List[Any]] = None,
open_meteo_temps: Optional[List[Any]] = None,
open_meteo_peak: Optional[Any] = None,
) -> List[str]:
"""Resolve the local high-temperature window, preferring multi-model hourly consensus."""
multi_model = weather_data.get("multi_model") if isinstance(weather_data, dict) else {}
if isinstance(multi_model, dict):
hourly_times = multi_model.get("hourly_times") or []
hourly_forecasts = multi_model.get("hourly_forecasts") or {}
if isinstance(hourly_forecasts, dict) and hourly_times:
hourly_values: List[Tuple[str, float]] = []
for idx, raw_time in enumerate(hourly_times):
t_str = str(raw_time or "")
if not t_str.startswith(local_date_str) or "T" not in t_str:
continue
time_part = t_str.split("T", 1)[1][:5]
try:
hour = int(time_part[:2])
except Exception:
continue
if not 8 <= hour <= 19:
continue
values = []
for model_name, series in hourly_forecasts.items():
if _is_excluded_model_name(model_name):
continue
if not isinstance(series, (list, tuple)) or idx >= len(series):
continue
value = _sf(series[idx])
if value is not None:
values.append(value)
median_value = _median(values)
if median_value is not None:
hourly_values.append((time_part, median_value))
peak_hours = _peak_hours_from_hourly_values(hourly_values)
if peak_hours:
return peak_hours
om_peak = _sf(open_meteo_peak)
if open_meteo_times and open_meteo_temps and om_peak is not None:
peak_hours = []
for t_raw, temp_raw in zip(open_meteo_times, open_meteo_temps):
t_str = str(t_raw or "")
temp = _sf(temp_raw)
if temp is None or not t_str.startswith(local_date_str) or "T" not in t_str:
continue
time_part = t_str.split("T", 1)[1][:5]
try:
hour = int(time_part[:2])
except Exception:
continue
if 8 <= hour <= 19 and abs(temp - om_peak) <= 0.2:
peak_hours.append(time_part)
return peak_hours
return []
def _resolve_settlement_source_label(city_name: Optional[str]) -> str:
if not city_name:
return "METAR"
@@ -420,16 +505,16 @@ def analyze_weather_trend(
is_cooling = trend_direction == "falling"
om_today = daily.get("temperature_2m_max", [None])[0]
om_today = _sf(current_forecasts.get("Open-Meteo"))
# === Peak hours ===
peak_hours = []
if times and temps and om_today is not None:
for t_str, temp in zip(times, temps):
if t_str.startswith(local_date_str) and abs(temp - om_today) <= 0.2:
hour = int(t_str.split("T")[1][:2])
if 8 <= hour <= 19:
peak_hours.append(t_str.split("T")[1][:5])
peak_hours = _resolve_peak_hours(
weather_data,
local_date_str,
times,
temps,
om_today,
)
if peak_hours:
first_peak_h = int(peak_hours[0].split(":")[0])
last_peak_h = int(peak_hours[-1].split(":")[0])