Skip to content

Commit 19b4ffc

Browse files
shrutipatel31facebook-github-bot
authored andcommitted
Add predictable metrics args to OverviewAnalysis (facebook#4711)
Summary: Adds configurable model_fit_threshold parameter to OverviewAnalysis for customizing the PredictableMetricsAnalysis healthcheck threshold. Reviewed By: bernardbeckerman Differential Revision: D89793977 Privacy Context Container: L1307644
1 parent 6216c4c commit 19b4ffc

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

ax/analysis/overview.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -105,6 +105,7 @@ def __init__(
105105
should_generate_reason: str | None = None,
106106
options: OrchestratorOptions | None = None,
107107
tier_metadata: dict[str, Any] | None = None,
108+
model_fit_threshold: float | None = None,
108109
) -> None:
109110
super().__init__()
110111
self.can_generate = can_generate
@@ -114,6 +115,7 @@ def __init__(
114115
self.should_generate_reason = should_generate_reason
115116
self.options = options
116117
self.tier_metadata = tier_metadata
118+
self.model_fit_threshold = model_fit_threshold
117119

118120
@override
119121
def validate_applicable_state(
@@ -197,7 +199,11 @@ def compute(
197199
if self.options is not None
198200
else None,
199201
ConstraintsFeasibilityAnalysis(),
200-
PredictableMetricsAnalysis(),
202+
PredictableMetricsAnalysis()
203+
if self.model_fit_threshold is None
204+
else PredictableMetricsAnalysis(
205+
model_fit_threshold=self.model_fit_threshold
206+
),
201207
BaselineImprovementAnalysis() if not has_batch_trials else None,
202208
*[
203209
SearchSpaceAnalysis(trial_index=trial.index)

0 commit comments

Comments
 (0)