Install UV package manager:
# macOS/Linux
curl -LsSf https://astral.sh/uv/install.sh | sh
# Windows
powershell -c "irm https://astral.sh/uv/install.ps1 | iex"- Clone the repository and checkout the feature branch:
git clone -----Anonimized-----
cd ARK- Create virtual environment and install dependencies:
uv venv
source .venv/bin/activate # On Windows: .venv\Scripts\activate
uv pip install -e .- Download IntelliGraphs datasets:
python -m intelligraphs.data_loaders.downloadCreate a configuration file in configs/ directory and run:
python -m kgvae.experiments.train --config configs/your_config.yamlThe training script automatically logs metrics to Weights & Biases. Configure with:
python -m kgvae.experiments.train \
--config configs/your_config.yaml \
--wandb-project your-project-name \
--wandb-entity your-entity- PyTorch >= 2.0.0
- IntelliGraphs >= 1.0.18
- Weights & Biases
- NumPy, scikit-learn, PyYAML, tqdm, matplotlib, pandas