README.md

---
library_name: gridfm-graphkit
pipeline_tag: tabular-regression
license: apache-2.0
tags:
  - power-systems
  - optimal-power-flow
  - genco
  - opfdata
---

GENCO optimal-power-flow checkpoints on OPFData ([paper](https://arxiv.org/abs/2608.09921)): IEEE 14, 30, 57, and 118, and GOC 500 and 2000, seeds 42, 3, and 17.

Evaluate on the paper branch by following [GENCO §5.2](https://github.com/gridfm/gridfm-paper-genco/blob/main/repro_instructions/5.2_OPF_on_opfdata.md). That guide uses graphkit [genco-paper-repro](https://github.com/gridfm/gridfm-graphkit/tree/genco-paper-repro) and is how the paper numbers were produced.

Or evaluate the same checkpoints on graphkit [main](https://github.com/gridfm/gridfm-graphkit/tree/main). `main` is under development, so the same numbers are not guaranteed. They matched the paper-branch result at commit [`f104900`](https://github.com/gridfm/gridfm-graphkit/commit/f104900245cba26d1380037eea562cf633a97484) (1 October 2026).

```bash
git clone https://github.com/gridfm/gridfm-graphkit.git
cd gridfm-graphkit
pip install -e .
pip install "huggingface_hub[cli]"
TORCH_CUDA_VERSION=$(python -c "import torch; print(torch.__version__ + ('+cpu' if torch.version.cuda is None else ''))")
pip install torch-scatter -f https://data.pyg.org/whl/torch-${TORCH_CUDA_VERSION}.html
export MLFLOW_ALLOW_FILE_STORE=true

mkdir -p data/case118_ieee/raw scripts/opfdata/splits
hf download gridfm/opfdata_case118_ieee --repo-type dataset --local-dir data/case118_ieee/raw
hf download gridfm/genco-opfdata --include "splits/*" --include "case118_ieee/seed42/**" --local-dir genco-opfdata
cp genco-opfdata/splits/*.pt scripts/opfdata/splits/

gridfm_graphkit evaluate \
  --config_conversion auto_inline \
  --config genco-opfdata/case118_ieee/seed42/HGNSQ_penalty_11_OPFData_case118_default.yaml \
  --data_path data \
  --model_path genco-opfdata/case118_ieee/seed42/best_model_state_dict.pt \
  --normalizer_stats genco-opfdata/case118_ieee/seed42/normalizer_stats.pt \
  --batch_size 512
```