Spaces:
Sleeping
Sleeping
Upload forecaster_cli.py
Browse files- forecaster_cli.py +9 -1
forecaster_cli.py
CHANGED
|
@@ -159,13 +159,21 @@ def get_prediction(ticker):
|
|
| 159 |
decision = "BUY" if prob > 0.65 else "PASS"
|
| 160 |
reasoning = generate_reasoning(features, tier, prob)
|
| 161 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 162 |
return {
|
| 163 |
"Ticker": ticker,
|
| 164 |
"Tier": tier,
|
| 165 |
"Decision": decision,
|
| 166 |
"Confidence": prob,
|
| 167 |
"Reasoning": reasoning,
|
| 168 |
-
"Features":
|
| 169 |
}
|
| 170 |
|
| 171 |
def run_inference(ticker):
|
|
|
|
| 159 |
decision = "BUY" if prob > 0.65 else "PASS"
|
| 160 |
reasoning = generate_reasoning(features, tier, prob)
|
| 161 |
|
| 162 |
+
import math
|
| 163 |
+
clean_features = {}
|
| 164 |
+
for k, v in features.items():
|
| 165 |
+
if isinstance(v, float) and math.isnan(v):
|
| 166 |
+
clean_features[k] = None
|
| 167 |
+
else:
|
| 168 |
+
clean_features[k] = v
|
| 169 |
+
|
| 170 |
return {
|
| 171 |
"Ticker": ticker,
|
| 172 |
"Tier": tier,
|
| 173 |
"Decision": decision,
|
| 174 |
"Confidence": prob,
|
| 175 |
"Reasoning": reasoning,
|
| 176 |
+
"Features": clean_features
|
| 177 |
}
|
| 178 |
|
| 179 |
def run_inference(ticker):
|