encoder_config.json

{
  "model_type": "wikimovies-transe-rgcn",
  "version": "v1-experimental",
  "description": "Frozen 2-layer R-GCN on WikiMovies KG with TransE initialization.",
  "embedding_dim": 128,
  "num_entities": 43234,
  "num_relations": 9,
  "frozen": true,
  "transe_init": true,
  "checkpoint_file": "wikimovies_rgcn.pt",
  "graph_align_repo": "https://github.com/riverlab/graph_int_LLM",
  "dataset": "MetaQA-3hop + WikiMovies KB (camazlucas/MetaQA)",
  "status": "experimental_baseline",
  "expected_updates": "This checkpoint is an early experiment baseline and will likely be superseded.",
  "rgcn_config": {
    "hidden_dim": 128,
    "num_layers": 2,
    "num_bases": 4,
    "learning_rate": 0.001,
    "batch_size": 1024,
    "num_epochs": 50,
    "num_negs_per_pos": 10,
    "dropout": 0.2,
    "split_ratios": [
      0.9,
      0.05,
      0.05
    ],
    "seed": 42,
    "freeze_transe_init": false
  }
}