This project uses a Graph Attention Network (GAT) to predict voltage magnitude and angle in a power network.
-
Clone the repository:
git clone https://github.com/alexisaglar/GAT_power_system.git -
Install the dependencies:
pip install -r requirements.txt -
Place your data in the
data/rawdirectory and run the preprocessing script (if needed).
-
Train the model:
python src/main.py -
Evaluate the model:
python src/evaluate.py
raw_data/: Contains raw and processed data.src/: Contains source code for data loading, model definition, training, and evaluation.plots/: Contains plots generated during network generation and results.requirements.txt: Lists the dependencies needed to run the project.README.md: Provides an overview and instructions for the project.