DHDRL commited on
Commit
2c1729d
·
verified ·
1 Parent(s): b6a0f69

Update zone_observation.py

Browse files
Files changed (1) hide show
  1. zone_observation.py +6 -0
zone_observation.py CHANGED
@@ -167,6 +167,7 @@ class ForecastConfig:
167
  clean_episode_ratio: float = 0.7
168
  event_spatial_correlation: float = 0.85
169
  shuffle_zone_order: bool = True
 
170
  alert_value_range: Tuple[float, float] = (0.8, 1.2)
171
  miss_penalty_range: Tuple[float, float] = (0.9, 1.1)
172
 
@@ -243,6 +244,10 @@ class ForecastConfig:
243
  raise ValueError(
244
  f"ForecastConfig: real_data_ratio={self.real_data_ratio} must be in [0, 1]"
245
  )
 
 
 
 
246
  if not (0.0 <= self.era5_ratio <= 1.0):
247
  raise ValueError(
248
  f"ForecastConfig: era5_ratio={self.era5_ratio} must be in [0, 1]"
@@ -320,6 +325,7 @@ class ForecastConfig:
320
  "clean_episode_ratio": self.clean_episode_ratio,
321
  "event_spatial_correlation": self.event_spatial_correlation,
322
  "shuffle_zone_order": self.shuffle_zone_order,
 
323
  "alert_value_range": list(self.alert_value_range),
324
  "miss_penalty_range": list(self.miss_penalty_range),
325
  "soft_reset": self.soft_reset,
 
167
  clean_episode_ratio: float = 0.7
168
  event_spatial_correlation: float = 0.85
169
  shuffle_zone_order: bool = True
170
+ forecast_flag_error_rate: float = 0.0
171
  alert_value_range: Tuple[float, float] = (0.8, 1.2)
172
  miss_penalty_range: Tuple[float, float] = (0.9, 1.1)
173
 
 
244
  raise ValueError(
245
  f"ForecastConfig: real_data_ratio={self.real_data_ratio} must be in [0, 1]"
246
  )
247
+ if not (0.0 <= self.forecast_flag_error_rate <= 1.0):
248
+ raise ValueError(
249
+ f'ForecastConfig: forecast_flag_error_rate={self.forecast_flag_error_rate} must be in [0, 1]'
250
+ )
251
  if not (0.0 <= self.era5_ratio <= 1.0):
252
  raise ValueError(
253
  f"ForecastConfig: era5_ratio={self.era5_ratio} must be in [0, 1]"
 
325
  "clean_episode_ratio": self.clean_episode_ratio,
326
  "event_spatial_correlation": self.event_spatial_correlation,
327
  "shuffle_zone_order": self.shuffle_zone_order,
328
+ "forecast_flag_error_rate": self.forecast_flag_error_rate,
329
  "alert_value_range": list(self.alert_value_range),
330
  "miss_penalty_range": list(self.miss_penalty_range),
331
  "soft_reset": self.soft_reset,