Skip to content

processing module

Jump to bottom
Syed Ibrahim Omer edited this page Apr 12, 2026 · 5 revisions

Processing Module

src/processing/ — transforms raw data into model-ready formats.

Files

File Purpose Key Functions/Classes
parse.py PDF visa table extraction PyMuPDF fitz, ProcessPoolExecutor, VISA_MAP normalization
news.py JSON articles → Parquet discover_country_directories(), load_and_flatten_country_articles(), tokenize_batch()
merge.py Consolidate encounter CSVs Column mapping, Polars lazy frame concatenation
build_panel.py Multi-source panel construction Lag/lead features, forward-fill, monthly × country merge
summarize.py FLAN-T5 cluster labeling NewsArticleSummarizer class, batch processing, stats tracking
run_summarization.py CLI orchestrator Argument validation, dry-run, stats-only modes
prompts.py Prompt templates PromptTemplate class, SUMMARIZATION_PROMPT, EXTRACTION_PROMPT, EVENTS_FOCUSED_PROMPT
utils.py Shared utilities setup_logger(), get_optimal_process_count(), MONTHS_MAP, VISA_MAP
maps.json Lookup tables Country name normalization, visa class mapping, month abbreviations

Data Flow

parse.py:       data/raw/visa/pdf/       → data/processed/visa_master.parquet
news.py:        data/raw/news/           → data/processed/news/news_*.parquet
merge.py:       data/raw/encounter/*.csv → merged encounter DataFrame
build_panel.py: All processed sources    → data/processed/train_panel.parquet
summarize.py:   news articles + TRT engine → articles with summary_t5 field

Key Design Patterns

  • Polars lazy frames: Memory-efficient operations on large datasets
  • ProcessPoolExecutor: Parallel PDF parsing bypassing GIL
  • Configurable prompts: PromptTemplate with max_input/max_output tokens
  • Checkpoint-friendly: Summarization tracks processed/skipped/error counts

See Also

Wiki navigation

Quick Start

  • Project Overview — Goals, research questions, methodology, and team
  • Glossary — Key terms used throughout this wiki

Data Sources

Raw inputs that feed the prediction system.

Page Description
Visa Data US Department of State visa issuance statistics (108 monthly PDFs)
Encounter Data CBP Southwest border encounter statistics (FY2019–2026)
Google News 170K+ news articles across 15 countries × 8 topics
Google Trends Monthly search-interest time series (15 countries × 8 keywords)
Exchange Rates IMF Real Effective Exchange Rate for 6 countries

Pipeline

The end-to-end flow from raw data to production forecasts.

Page Description
Data Collection Ingestion layer: async scraping, bounded concurrency, retry logic
Data Processing PDF parsing, JSON→Parquet, encounter merging
NLP Enrichment Embedding → Clustering → Labeling → Sentiment
Panel Construction Feature engineering: 18 lag features, 6 lead targets
Training Pipeline Out-of-time train/test split, 4 architectures
Inference Pipeline Horizon-aware ensemble, production prediction flow

Models

Machine learning architectures and their roles in the ensemble.

Page Description
Random Forest cuML GPU Random Forest — best at short horizons (Lead 1–2)
LSTM MigrationLSTM — country-aware with SurgeJointLoss
Transformer MigrationTransformer — best at long horizons (Lead 5–6)
Horizon-Aware Ensemble Dynamic weighting: RF→short, Transformer→long
SurgeJointLoss Dual-objective loss: Huber + BCE for crisis detection
Jina v5 Embeddings TensorRT INT8 news article embeddings (768-dim)
Flan-T5 Summarization TensorRT INT8 cluster labeling engine

Analysis Methods

Statistical techniques driving the lead-lag and surge analysis.

Page Description
Lead-Lag Analysis Pearson correlation at 0–6 month offsets
Surge Detection Quantile-based and σ-threshold spike identification
Sentiment Analysis Rule-based lexicon scoring for migration-relevant news
Event Clustering HDBSCAN GPU clustering + LED label generation
Cross-Correlation Analysis CCF analysis, VAR benchmarking, ADF stationarity tests
Multiple Comparison Correction Benjamini-Hochberg FDR for 58 significant signals

Key Findings

What the system discovered about migration predictability.

Page Description
Event-Visa Findings News events as leading indicators (r=0.617 at 3-month lag)
Exchange Rate Findings Exchange rate signals (DR r=0.498 at 2-month lag)
Model Performance Ensemble results: F1=0.96 at Lead 1, F1=0.86 at Lead 6

Source Modules

Reference documentation for every src/ subpackage and key files.

Page Description
Main Entry Point src/main.py CLI: bootstrap, collect-live, sync-data
Collection Module src/collection/* — visa, encounter, news, trends, HF sync
Processing Module src/processing/* — parse, merge, build_panel, summarize
Analysis Module src/analysis/* — events, exchange_rate, trends_analysis, plots
Models Module src/models/* — surge_model, train_and_evaluate, inference
News Scraper Deep dive: batch decoding, checkpoint recovery, throttling
PDF Parser Deep dive: PyMuPDF table extraction, VISA_MAP normalization
TensorRT Engines Deep dive: Jina-v5, Flan-T5, LED TensorRT engines
Build Panel Detail Deep dive: lag/lead construction, forward-fill strategies
HF Sync Deep dive: bidirectional Hugging Face Hub sync

Infrastructure

Compute, reproducibility, and operational details.

Page Description
GPU Acceleration TensorRT INT8, cuML, CUDA streams, NVML profiling
Reproducibility HF bootstrap, run.sh pipeline, dependency checking

Clone this wiki locally