Spaces:
Sleeping
Sleeping
File size: 1,618 Bytes
f87e795 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 | """
main.py
-------
Entry point for SchemeImpactNet.
Usage:
python main.py # Stage 1 — Maharashtra
python main.py --stage 2 # Stage 2 — All-India
python main.py --stage 3 # Stage 3 — All-India + optimize
python main.py --stage 3 --state Maharashtra # Stage 3, one state
python main.py --optimize-only # Run optimizer on existing predictions
"""
import sys
from src.pipeline import run_pipeline, run_optimizer_step
if __name__ == "__main__":
args = sys.argv[1:]
stage = 1
if "--stage" in args:
stage = int(args[args.index("--stage") + 1])
scope_state = None
if "--state" in args:
scope_state = args[args.index("--state") + 1]
optimize_only = "--optimize-only" in args
if optimize_only:
print("\nRunning optimizer on existing predictions...")
run_optimizer_step(scope_state=scope_state)
else:
predictions = run_pipeline(stage=stage)
print(f"\nTop 10 predicted districts (2023):")
latest = predictions[predictions["financial_year"] == 2023]
print(
latest[["state", "district", "person_days_lakhs", "predicted_persondays"]]
.sort_values("predicted_persondays", ascending=False)
.head(10)
.to_string(index=False)
)
# Stage 3: automatically run optimizer after model
if stage == 3:
print("\n" + "─" * 60)
print(" Running Stage 3 Budget Optimizer...")
print("─" * 60)
run_optimizer_step(scope_state=scope_state)
|