balance_diagnostics
Check covariate balance after matching or weighting, estimate propensity scores if needed, and get violation flags plus next steps for causal analysis.
Instructions
Unified balance diagnostics for matching and weighting estimators.
Input Schema
| Name | Required | Description | Default |
|---|---|---|---|
| ps | No | Propensity scores. If omitted, estimated with ``method``. | |
| detail | No | Payload depth: 'minimal' (~150 tokens) for sub-step calls where only the point estimate is needed; 'standard' (~1K tokens) for diagnostics + coefficient table; 'agent' (~2K tokens, default) adds violations / next_steps / suggested_functions so the LLM can plan its next call without another round-trip. | agent |
| method | No | Propensity-score model when ``ps`` is not supplied. | logit |
| weights | No | Observation weights after matching/weighting. If omitted, ATE inverse-propensity weights are computed from ``ps``. | |
| as_handle | No | If true, cache the fitted result on the server and return result_id + result_uri alongside the JSON payload so a subsequent tools/call can chain without re-running. | |
| data_path | Yes | Absolute path or URL to a data file. Supported: .csv / .tsv / .txt (delimited), .parquet / .pq, .feather / .arrow, .xlsx / .xls, .dta (Stata), .json / .jsonl. Schemes: file://, s3://, gs://, https://. | |
| result_id | No | Optional handle to a previously-fitted result (returned by an earlier call when as_handle=true). Tools that operate on a fitted object accept this in place of re-supplying data_path + columns. | |
| threshold | No | Balance threshold for absolute standardized mean differences. | |
| treatment | Yes | Binary treatment indicator. | |
| covariates | Yes | Covariates to audit. | |
| data_columns | No | Optional column projection. Parquet/Feather/Stata loaders honour this for fast partial reads. | |
| data_sample_n | No | Optional uniform random subsample size (seed=0, deterministic) — useful on huge panels. |
Output Schema
| Name | Required | Description | Default |
|---|---|---|---|
No arguments | |||