utils/preprocessing.py

def extract_features(agent_data):
    return [
        agent_data.get("weekly_calls", 0),
        agent_data.get("missed_visits", 0),
        agent_data.get("travel_distance", 0),
        agent_data.get("lead_drop_rate", 0.0)
    ]