KG-to-KG Semantic Similarity Baselines

This repository contains baseline code for the KG-to-KG Semantic Similarity benchmark dataset:

https://huggingface.co/datasets/seungryeol-22/KG-to-KG-Semantic-Similarity

The code preserves the original baseline pipeline:

scoring -> ranking -> metric

The original scripts expect legacy local input folders such as:

cc_news_graph/
wikitext_graph_data/
cc_news_graph_verbalized/
wikitext_graph_data_verbalized/

The Hugging Face dataset stores the same graph pairs as Parquet files. Run the materialization script first to recreate the legacy folders from the dataset repository.

Prepare Data

python scripts/materialize_legacy_dataset.py --output . --force

This downloads seungryeol-22/KG-to-KG-Semantic-Similarity and creates:

cc_news_graph/
wikitext_graph_data/
cc_news_graph_verbalized/
wikitext_graph_data_verbalized/

Run Pipeline

Run a scoring script:

python scoring/base-kernel-scoring.py

Then run ranking:

python evaluation/ranking.py

Then run metrics:

python evaluation/metric.py

Results are written by the original scripts under Result/ and Metric_Results/.

Code Layout

scoring/
evaluation/
scripts/materialize_legacy_dataset.py

Dependencies

Install the packages needed by the baseline you plan to run. The adapter requires:

pip install datasets

Graph-kernel baselines require:

pip install grakel tqdm numpy

SBERT baseline requires:

pip install sentence-transformers tqdm numpy

KGE and InGram baselines require PyTorch-related dependencies such as torch, torch-geometric, scipy, scikit-learn, and igraph.

Paper and Dataset

Paper: https://arxiv.org/abs/2606.29180

Dataset: https://huggingface.co/datasets/seungryeol-22/KG-to-KG-Semantic-Similarity

Code source: https://github.com/SeungRyeolBaek/KG-to-KG-Semantic-Similarity

Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Dataset used to train seungryeol-22/KG-to-KG-Semantic-Similarity-Baselines

Paper for seungryeol-22/KG-to-KG-Semantic-Similarity-Baselines