├── Makefile ├── README.md ├── bin ├── eval.py ├── score.py └── trainer.py ├── configs ├── clip-graph-directed │ ├── inductive-causal │ │ ├── pubmed │ │ │ └── base.yaml │ │ ├── trex │ │ │ └── base.yaml │ │ └── twitter-small │ │ │ └── base.yaml │ └── inductive-masked │ │ ├── pubmed │ │ └── base.yaml │ │ ├── trex │ │ └── base.yaml │ │ └── twitter-small │ │ └── base.yaml ├── clip-graph │ ├── causal-lm-train │ │ ├── pubmed │ │ │ ├── base.yaml │ │ │ └── extra-epoch.yaml │ │ ├── trex │ │ │ ├── base.yaml │ │ │ └── extra-epoch.yaml │ │ └── twitter-small │ │ │ ├── base.yaml │ │ │ └── extra-epoch.yaml │ ├── inductive-causal │ │ ├── pubmed │ │ │ ├── base.yaml │ │ │ ├── sim-smoothed-exp-thick-tail.yaml │ │ │ ├── sim-smoothed-exp.yaml │ │ │ ├── sim-smoothed-identity-50.yaml │ │ │ └── sim-smoothed-identity.yaml │ │ ├── trex │ │ │ ├── base.yaml │ │ │ ├── sim-smoothed-exp-thick-tail.yaml │ │ │ ├── sim-smoothed-exp.yaml │ │ │ ├── sim-smoothed-identity-50.yaml │ │ │ └── sim-smoothed-identity.yaml │ │ └── twitter-small │ │ │ ├── base.yaml │ │ │ ├── sim-smoothed-exp-thick-tail.yaml │ │ │ ├── sim-smoothed-exp.yaml │ │ │ ├── sim-smoothed-identity-50.yaml │ │ │ └── sim-smoothed-identity.yaml │ └── inductive-masked │ │ ├── pubmed │ │ ├── base.yaml │ │ ├── sim-smoothed-exp-thick-tail.yaml │ │ ├── sim-smoothed-exp.yaml │ │ ├── sim-smoothed-identity-50.yaml │ │ └── sim-smoothed-identity.yaml │ │ ├── trex │ │ ├── base.yaml │ │ ├── sim-smoothed-exp-thick-tail.yaml │ │ ├── sim-smoothed-exp.yaml │ │ ├── sim-smoothed-identity-50.yaml │ │ └── sim-smoothed-identity.yaml │ │ └── twitter-small │ │ ├── base.yaml │ │ ├── sim-smoothed-exp-thick-tail.yaml │ │ ├── sim-smoothed-exp.yaml │ │ ├── sim-smoothed-identity-50.yaml │ │ └── sim-smoothed-identity.yaml ├── comparisons.yaml ├── eval-datasets │ ├── pubmed │ │ ├── causal-directed.yaml │ │ ├── causal.yaml │ │ ├── masked-directed.yaml │ │ └── masked.yaml │ ├── trex │ │ ├── causal-directed.yaml │ │ ├── causal.yaml │ │ ├── masked-directed.yaml │ │ └── masked.yaml │ └── twitter-small │ │ ├── causal-directed.yaml │ │ ├── causal.yaml │ │ ├── masked-directed.yaml │ │ └── masked.yaml ├── gnn-pretrain-directed │ ├── pubmed │ │ ├── base-lr-1e-2-20epoch.yaml │ │ ├── base-lr-1e-2-30epoch.yaml │ │ ├── base-lr-1e-3-20epoch.yaml │ │ └── base-lr-1e-3-30epoch.yaml │ ├── trex │ │ ├── base-lr-1e-2-20epoch.yaml │ │ ├── base-lr-1e-2-30epoch.yaml │ │ ├── base-lr-1e-3-20epoch.yaml │ │ └── base-lr-1e-3-30epoch.yaml │ └── twitter-small │ │ ├── base-lr-1e-2-20epoch.yaml │ │ ├── base-lr-1e-2-30epoch.yaml │ │ ├── base-lr-1e-3-20epoch.yaml │ │ └── base-lr-1e-3-30epoch.yaml ├── gnn-pretrain-textinit-directed │ ├── pubmed │ │ ├── base-lr-1e-2-20epoch.yaml │ │ ├── base-lr-1e-2-30epoch.yaml │ │ ├── base-lr-1e-3-20epoch.yaml │ │ └── base-lr-1e-3-30epoch.yaml │ ├── trex │ │ ├── base-lr-1e-2-20epoch.yaml │ │ ├── base-lr-1e-2-30epoch.yaml │ │ ├── base-lr-1e-3-20epoch.yaml │ │ └── base-lr-1e-3-30epoch.yaml │ └── twitter-small │ │ ├── base-lr-1e-2-20epoch.yaml │ │ ├── base-lr-1e-2-30epoch.yaml │ │ ├── base-lr-1e-3-20epoch.yaml │ │ └── base-lr-1e-3-30epoch.yaml ├── gnn-pretrain-textinit │ ├── pubmed │ │ ├── base-lr-1e-2-20epoch.yaml │ │ ├── base-lr-1e-2-30epoch.yaml │ │ ├── base-lr-1e-3-20epoch.yaml │ │ └── base-lr-1e-3-30epoch.yaml │ ├── trex │ │ ├── base-lr-1e-2-20epoch.yaml │ │ ├── base-lr-1e-2-30epoch.yaml │ │ ├── base-lr-1e-3-20epoch.yaml │ │ └── base-lr-1e-3-30epoch.yaml │ └── twitter-small │ │ ├── base-lr-1e-2-20epoch.yaml │ │ ├── base-lr-1e-2-30epoch.yaml │ │ ├── base-lr-1e-3-20epoch.yaml │ │ └── base-lr-1e-3-30epoch.yaml ├── gnn-pretrain │ ├── pubmed │ │ ├── base-lr-1e-2-20epoch.yaml │ │ ├── base-lr-1e-2-30epoch.yaml │ │ ├── base-lr-1e-3-20epoch.yaml │ │ └── base-lr-1e-3-30epoch.yaml │ ├── trex │ │ ├── base-lr-1e-2-20epoch.yaml │ │ ├── base-lr-1e-2-30epoch.yaml │ │ ├── base-lr-1e-3-20epoch.yaml │ │ └── base-lr-1e-3-30epoch.yaml │ └── twitter-small │ │ ├── base-lr-1e-2-20epoch.yaml │ │ ├── base-lr-1e-2-30epoch.yaml │ │ ├── base-lr-1e-3-20epoch.yaml │ │ └── base-lr-1e-3-30epoch.yaml └── lm-pretrain │ ├── pubmed │ ├── causal.yaml │ └── masked.yaml │ ├── trex │ ├── causal.yaml │ └── masked.yaml │ └── twitter-small │ ├── causal.yaml │ └── masked.yaml ├── data ├── .gitignore ├── dataset-stats.csv ├── evals │ ├── pubmed-directed-test-causal-base.json │ ├── pubmed-directed-test-masked-base.json │ ├── pubmed-test-causal-base.json │ ├── pubmed-test-causal-sim10.json │ ├── pubmed-test-masked-base.json │ ├── pubmed-test-masked-sim10.json │ ├── trex-directed-test-causal-base.json │ ├── trex-directed-test-masked-base.json │ ├── trex-test-causal-base.json │ ├── trex-test-causal-sim10.json │ ├── trex-test-masked-base.json │ ├── trex-test-masked-sim10.json │ ├── twitter-small-directed-test-causal-base.json │ ├── twitter-small-directed-test-masked-base.json │ ├── twitter-small-test-causal-base.json │ ├── twitter-small-test-causal-sim10.json │ ├── twitter-small-test-masked-base.json │ └── twitter-small-test-masked-sim10.json ├── link-prediction-eval.csv ├── pubmed-textinit │ └── raw ├── pubmed │ ├── .gitignore │ └── prep.py ├── trex-textinit │ └── raw ├── trex │ ├── .gitignore │ ├── orig │ │ └── README.md │ └── prep.py ├── twitter-small-textinit │ ├── derived │ └── raw └── twitter-small │ └── 1-node-properties.py ├── notebooks ├── .ipynb_checkpoints │ ├── 1a-datasets-checkpoint.ipynb │ ├── 1b-save-splits-checkpoint.ipynb │ ├── 2a-eval-cross-checkpoint.ipynb │ ├── 2a-eval-text-pubmed-checkpoint.ipynb │ ├── 2b-eval-text-trex-checkpoint.ipynb │ ├── 2c-eval-text-twitter-small-checkpoint.ipynb │ ├── 3a-eval-link-pred-checkpoint.ipynb │ ├── 4a-eval-descriptive-pubmed-checkpoint.ipynb │ ├── 4b-eval-descriptive-trex-checkpoint.ipynb │ ├── 4c-eval-descriptive-twitter-small-checkpoint.ipynb │ └── 5a-eval-cross-checkpoint.ipynb ├── 1a-datasets.ipynb ├── 1b-save-splits.ipynb ├── 2a-eval-text-pubmed.ipynb ├── 2b-eval-text-trex.ipynb ├── 2c-eval-text-twitter-small.ipynb ├── 3a-eval-link-pred.ipynb ├── 4a-eval-descriptive-pubmed.ipynb ├── 4b-eval-descriptive-trex.ipynb ├── 4c-eval-descriptive-twitter-small.ipynb ├── 5a-eval-cross.ipynb ├── lr-tuning.ipynb └── utils.py ├── pyproject.toml ├── setup.cfg ├── setup.py └── src └── clip_graph ├── __init__.py ├── data ├── __init__.py ├── callbacks.py ├── datamodule.py └── dataset.py ├── eval.py ├── lit.py ├── losses.py ├── models.py ├── optimizers.py ├── scoring.py └── utils.py /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/Makefile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/README.md -------------------------------------------------------------------------------- /bin/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/bin/eval.py -------------------------------------------------------------------------------- /bin/score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/bin/score.py -------------------------------------------------------------------------------- /bin/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/bin/trainer.py -------------------------------------------------------------------------------- /configs/clip-graph-directed/inductive-causal/pubmed/base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/clip-graph-directed/inductive-causal/pubmed/base.yaml -------------------------------------------------------------------------------- /configs/clip-graph-directed/inductive-causal/trex/base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/clip-graph-directed/inductive-causal/trex/base.yaml -------------------------------------------------------------------------------- /configs/clip-graph-directed/inductive-causal/twitter-small/base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/clip-graph-directed/inductive-causal/twitter-small/base.yaml -------------------------------------------------------------------------------- /configs/clip-graph-directed/inductive-masked/pubmed/base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/clip-graph-directed/inductive-masked/pubmed/base.yaml -------------------------------------------------------------------------------- /configs/clip-graph-directed/inductive-masked/trex/base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/clip-graph-directed/inductive-masked/trex/base.yaml -------------------------------------------------------------------------------- /configs/clip-graph-directed/inductive-masked/twitter-small/base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/clip-graph-directed/inductive-masked/twitter-small/base.yaml -------------------------------------------------------------------------------- /configs/clip-graph/causal-lm-train/pubmed/base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/clip-graph/causal-lm-train/pubmed/base.yaml -------------------------------------------------------------------------------- /configs/clip-graph/causal-lm-train/pubmed/extra-epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/clip-graph/causal-lm-train/pubmed/extra-epoch.yaml -------------------------------------------------------------------------------- /configs/clip-graph/causal-lm-train/trex/base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/clip-graph/causal-lm-train/trex/base.yaml -------------------------------------------------------------------------------- /configs/clip-graph/causal-lm-train/trex/extra-epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/clip-graph/causal-lm-train/trex/extra-epoch.yaml -------------------------------------------------------------------------------- /configs/clip-graph/causal-lm-train/twitter-small/base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/clip-graph/causal-lm-train/twitter-small/base.yaml -------------------------------------------------------------------------------- /configs/clip-graph/causal-lm-train/twitter-small/extra-epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/clip-graph/causal-lm-train/twitter-small/extra-epoch.yaml -------------------------------------------------------------------------------- /configs/clip-graph/inductive-causal/pubmed/base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/clip-graph/inductive-causal/pubmed/base.yaml -------------------------------------------------------------------------------- /configs/clip-graph/inductive-causal/pubmed/sim-smoothed-exp-thick-tail.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/clip-graph/inductive-causal/pubmed/sim-smoothed-exp-thick-tail.yaml -------------------------------------------------------------------------------- /configs/clip-graph/inductive-causal/pubmed/sim-smoothed-exp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/clip-graph/inductive-causal/pubmed/sim-smoothed-exp.yaml -------------------------------------------------------------------------------- /configs/clip-graph/inductive-causal/pubmed/sim-smoothed-identity-50.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/clip-graph/inductive-causal/pubmed/sim-smoothed-identity-50.yaml -------------------------------------------------------------------------------- /configs/clip-graph/inductive-causal/pubmed/sim-smoothed-identity.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/clip-graph/inductive-causal/pubmed/sim-smoothed-identity.yaml -------------------------------------------------------------------------------- /configs/clip-graph/inductive-causal/trex/base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/clip-graph/inductive-causal/trex/base.yaml -------------------------------------------------------------------------------- /configs/clip-graph/inductive-causal/trex/sim-smoothed-exp-thick-tail.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/clip-graph/inductive-causal/trex/sim-smoothed-exp-thick-tail.yaml -------------------------------------------------------------------------------- /configs/clip-graph/inductive-causal/trex/sim-smoothed-exp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/clip-graph/inductive-causal/trex/sim-smoothed-exp.yaml -------------------------------------------------------------------------------- /configs/clip-graph/inductive-causal/trex/sim-smoothed-identity-50.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/clip-graph/inductive-causal/trex/sim-smoothed-identity-50.yaml -------------------------------------------------------------------------------- /configs/clip-graph/inductive-causal/trex/sim-smoothed-identity.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/clip-graph/inductive-causal/trex/sim-smoothed-identity.yaml -------------------------------------------------------------------------------- /configs/clip-graph/inductive-causal/twitter-small/base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/clip-graph/inductive-causal/twitter-small/base.yaml -------------------------------------------------------------------------------- /configs/clip-graph/inductive-causal/twitter-small/sim-smoothed-exp-thick-tail.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/clip-graph/inductive-causal/twitter-small/sim-smoothed-exp-thick-tail.yaml -------------------------------------------------------------------------------- /configs/clip-graph/inductive-causal/twitter-small/sim-smoothed-exp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/clip-graph/inductive-causal/twitter-small/sim-smoothed-exp.yaml -------------------------------------------------------------------------------- /configs/clip-graph/inductive-causal/twitter-small/sim-smoothed-identity-50.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/clip-graph/inductive-causal/twitter-small/sim-smoothed-identity-50.yaml -------------------------------------------------------------------------------- /configs/clip-graph/inductive-causal/twitter-small/sim-smoothed-identity.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/clip-graph/inductive-causal/twitter-small/sim-smoothed-identity.yaml -------------------------------------------------------------------------------- /configs/clip-graph/inductive-masked/pubmed/base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/clip-graph/inductive-masked/pubmed/base.yaml -------------------------------------------------------------------------------- /configs/clip-graph/inductive-masked/pubmed/sim-smoothed-exp-thick-tail.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/clip-graph/inductive-masked/pubmed/sim-smoothed-exp-thick-tail.yaml -------------------------------------------------------------------------------- /configs/clip-graph/inductive-masked/pubmed/sim-smoothed-exp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/clip-graph/inductive-masked/pubmed/sim-smoothed-exp.yaml -------------------------------------------------------------------------------- /configs/clip-graph/inductive-masked/pubmed/sim-smoothed-identity-50.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/clip-graph/inductive-masked/pubmed/sim-smoothed-identity-50.yaml -------------------------------------------------------------------------------- /configs/clip-graph/inductive-masked/pubmed/sim-smoothed-identity.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/clip-graph/inductive-masked/pubmed/sim-smoothed-identity.yaml -------------------------------------------------------------------------------- /configs/clip-graph/inductive-masked/trex/base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/clip-graph/inductive-masked/trex/base.yaml -------------------------------------------------------------------------------- /configs/clip-graph/inductive-masked/trex/sim-smoothed-exp-thick-tail.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/clip-graph/inductive-masked/trex/sim-smoothed-exp-thick-tail.yaml -------------------------------------------------------------------------------- /configs/clip-graph/inductive-masked/trex/sim-smoothed-exp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/clip-graph/inductive-masked/trex/sim-smoothed-exp.yaml -------------------------------------------------------------------------------- /configs/clip-graph/inductive-masked/trex/sim-smoothed-identity-50.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/clip-graph/inductive-masked/trex/sim-smoothed-identity-50.yaml -------------------------------------------------------------------------------- /configs/clip-graph/inductive-masked/trex/sim-smoothed-identity.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/clip-graph/inductive-masked/trex/sim-smoothed-identity.yaml -------------------------------------------------------------------------------- /configs/clip-graph/inductive-masked/twitter-small/base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/clip-graph/inductive-masked/twitter-small/base.yaml -------------------------------------------------------------------------------- /configs/clip-graph/inductive-masked/twitter-small/sim-smoothed-exp-thick-tail.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/clip-graph/inductive-masked/twitter-small/sim-smoothed-exp-thick-tail.yaml -------------------------------------------------------------------------------- /configs/clip-graph/inductive-masked/twitter-small/sim-smoothed-exp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/clip-graph/inductive-masked/twitter-small/sim-smoothed-exp.yaml -------------------------------------------------------------------------------- /configs/clip-graph/inductive-masked/twitter-small/sim-smoothed-identity-50.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/clip-graph/inductive-masked/twitter-small/sim-smoothed-identity-50.yaml -------------------------------------------------------------------------------- /configs/clip-graph/inductive-masked/twitter-small/sim-smoothed-identity.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/clip-graph/inductive-masked/twitter-small/sim-smoothed-identity.yaml -------------------------------------------------------------------------------- /configs/comparisons.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/comparisons.yaml -------------------------------------------------------------------------------- /configs/eval-datasets/pubmed/causal-directed.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/eval-datasets/pubmed/causal-directed.yaml -------------------------------------------------------------------------------- /configs/eval-datasets/pubmed/causal.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/eval-datasets/pubmed/causal.yaml -------------------------------------------------------------------------------- /configs/eval-datasets/pubmed/masked-directed.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/eval-datasets/pubmed/masked-directed.yaml -------------------------------------------------------------------------------- /configs/eval-datasets/pubmed/masked.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/eval-datasets/pubmed/masked.yaml -------------------------------------------------------------------------------- /configs/eval-datasets/trex/causal-directed.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/eval-datasets/trex/causal-directed.yaml -------------------------------------------------------------------------------- /configs/eval-datasets/trex/causal.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/eval-datasets/trex/causal.yaml -------------------------------------------------------------------------------- /configs/eval-datasets/trex/masked-directed.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/eval-datasets/trex/masked-directed.yaml -------------------------------------------------------------------------------- /configs/eval-datasets/trex/masked.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/eval-datasets/trex/masked.yaml -------------------------------------------------------------------------------- /configs/eval-datasets/twitter-small/causal-directed.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/eval-datasets/twitter-small/causal-directed.yaml -------------------------------------------------------------------------------- /configs/eval-datasets/twitter-small/causal.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/eval-datasets/twitter-small/causal.yaml -------------------------------------------------------------------------------- /configs/eval-datasets/twitter-small/masked-directed.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/eval-datasets/twitter-small/masked-directed.yaml -------------------------------------------------------------------------------- /configs/eval-datasets/twitter-small/masked.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/eval-datasets/twitter-small/masked.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain-directed/pubmed/base-lr-1e-2-20epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain-directed/pubmed/base-lr-1e-2-20epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain-directed/pubmed/base-lr-1e-2-30epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain-directed/pubmed/base-lr-1e-2-30epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain-directed/pubmed/base-lr-1e-3-20epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain-directed/pubmed/base-lr-1e-3-20epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain-directed/pubmed/base-lr-1e-3-30epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain-directed/pubmed/base-lr-1e-3-30epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain-directed/trex/base-lr-1e-2-20epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain-directed/trex/base-lr-1e-2-20epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain-directed/trex/base-lr-1e-2-30epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain-directed/trex/base-lr-1e-2-30epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain-directed/trex/base-lr-1e-3-20epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain-directed/trex/base-lr-1e-3-20epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain-directed/trex/base-lr-1e-3-30epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain-directed/trex/base-lr-1e-3-30epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain-directed/twitter-small/base-lr-1e-2-20epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain-directed/twitter-small/base-lr-1e-2-20epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain-directed/twitter-small/base-lr-1e-2-30epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain-directed/twitter-small/base-lr-1e-2-30epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain-directed/twitter-small/base-lr-1e-3-20epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain-directed/twitter-small/base-lr-1e-3-20epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain-directed/twitter-small/base-lr-1e-3-30epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain-directed/twitter-small/base-lr-1e-3-30epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain-textinit-directed/pubmed/base-lr-1e-2-20epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain-textinit-directed/pubmed/base-lr-1e-2-20epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain-textinit-directed/pubmed/base-lr-1e-2-30epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain-textinit-directed/pubmed/base-lr-1e-2-30epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain-textinit-directed/pubmed/base-lr-1e-3-20epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain-textinit-directed/pubmed/base-lr-1e-3-20epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain-textinit-directed/pubmed/base-lr-1e-3-30epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain-textinit-directed/pubmed/base-lr-1e-3-30epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain-textinit-directed/trex/base-lr-1e-2-20epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain-textinit-directed/trex/base-lr-1e-2-20epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain-textinit-directed/trex/base-lr-1e-2-30epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain-textinit-directed/trex/base-lr-1e-2-30epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain-textinit-directed/trex/base-lr-1e-3-20epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain-textinit-directed/trex/base-lr-1e-3-20epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain-textinit-directed/trex/base-lr-1e-3-30epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain-textinit-directed/trex/base-lr-1e-3-30epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain-textinit-directed/twitter-small/base-lr-1e-2-20epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain-textinit-directed/twitter-small/base-lr-1e-2-20epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain-textinit-directed/twitter-small/base-lr-1e-2-30epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain-textinit-directed/twitter-small/base-lr-1e-2-30epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain-textinit-directed/twitter-small/base-lr-1e-3-20epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain-textinit-directed/twitter-small/base-lr-1e-3-20epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain-textinit-directed/twitter-small/base-lr-1e-3-30epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain-textinit-directed/twitter-small/base-lr-1e-3-30epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain-textinit/pubmed/base-lr-1e-2-20epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain-textinit/pubmed/base-lr-1e-2-20epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain-textinit/pubmed/base-lr-1e-2-30epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain-textinit/pubmed/base-lr-1e-2-30epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain-textinit/pubmed/base-lr-1e-3-20epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain-textinit/pubmed/base-lr-1e-3-20epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain-textinit/pubmed/base-lr-1e-3-30epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain-textinit/pubmed/base-lr-1e-3-30epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain-textinit/trex/base-lr-1e-2-20epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain-textinit/trex/base-lr-1e-2-20epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain-textinit/trex/base-lr-1e-2-30epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain-textinit/trex/base-lr-1e-2-30epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain-textinit/trex/base-lr-1e-3-20epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain-textinit/trex/base-lr-1e-3-20epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain-textinit/trex/base-lr-1e-3-30epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain-textinit/trex/base-lr-1e-3-30epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain-textinit/twitter-small/base-lr-1e-2-20epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain-textinit/twitter-small/base-lr-1e-2-20epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain-textinit/twitter-small/base-lr-1e-2-30epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain-textinit/twitter-small/base-lr-1e-2-30epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain-textinit/twitter-small/base-lr-1e-3-20epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain-textinit/twitter-small/base-lr-1e-3-20epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain-textinit/twitter-small/base-lr-1e-3-30epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain-textinit/twitter-small/base-lr-1e-3-30epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain/pubmed/base-lr-1e-2-20epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain/pubmed/base-lr-1e-2-20epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain/pubmed/base-lr-1e-2-30epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain/pubmed/base-lr-1e-2-30epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain/pubmed/base-lr-1e-3-20epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain/pubmed/base-lr-1e-3-20epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain/pubmed/base-lr-1e-3-30epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain/pubmed/base-lr-1e-3-30epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain/trex/base-lr-1e-2-20epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain/trex/base-lr-1e-2-20epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain/trex/base-lr-1e-2-30epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain/trex/base-lr-1e-2-30epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain/trex/base-lr-1e-3-20epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain/trex/base-lr-1e-3-20epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain/trex/base-lr-1e-3-30epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain/trex/base-lr-1e-3-30epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain/twitter-small/base-lr-1e-2-20epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain/twitter-small/base-lr-1e-2-20epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain/twitter-small/base-lr-1e-2-30epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain/twitter-small/base-lr-1e-2-30epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain/twitter-small/base-lr-1e-3-20epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain/twitter-small/base-lr-1e-3-20epoch.yaml -------------------------------------------------------------------------------- /configs/gnn-pretrain/twitter-small/base-lr-1e-3-30epoch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/gnn-pretrain/twitter-small/base-lr-1e-3-30epoch.yaml -------------------------------------------------------------------------------- /configs/lm-pretrain/pubmed/causal.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/lm-pretrain/pubmed/causal.yaml -------------------------------------------------------------------------------- /configs/lm-pretrain/pubmed/masked.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/lm-pretrain/pubmed/masked.yaml -------------------------------------------------------------------------------- /configs/lm-pretrain/trex/causal.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/lm-pretrain/trex/causal.yaml -------------------------------------------------------------------------------- /configs/lm-pretrain/trex/masked.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/lm-pretrain/trex/masked.yaml -------------------------------------------------------------------------------- /configs/lm-pretrain/twitter-small/causal.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/lm-pretrain/twitter-small/causal.yaml -------------------------------------------------------------------------------- /configs/lm-pretrain/twitter-small/masked.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/configs/lm-pretrain/twitter-small/masked.yaml -------------------------------------------------------------------------------- /data/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/data/.gitignore -------------------------------------------------------------------------------- /data/dataset-stats.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/data/dataset-stats.csv -------------------------------------------------------------------------------- /data/evals/pubmed-directed-test-causal-base.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/data/evals/pubmed-directed-test-causal-base.json -------------------------------------------------------------------------------- /data/evals/pubmed-directed-test-masked-base.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/data/evals/pubmed-directed-test-masked-base.json -------------------------------------------------------------------------------- /data/evals/pubmed-test-causal-base.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/data/evals/pubmed-test-causal-base.json -------------------------------------------------------------------------------- /data/evals/pubmed-test-causal-sim10.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/data/evals/pubmed-test-causal-sim10.json -------------------------------------------------------------------------------- /data/evals/pubmed-test-masked-base.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/data/evals/pubmed-test-masked-base.json -------------------------------------------------------------------------------- /data/evals/pubmed-test-masked-sim10.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/data/evals/pubmed-test-masked-sim10.json -------------------------------------------------------------------------------- /data/evals/trex-directed-test-causal-base.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/data/evals/trex-directed-test-causal-base.json -------------------------------------------------------------------------------- /data/evals/trex-directed-test-masked-base.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/data/evals/trex-directed-test-masked-base.json -------------------------------------------------------------------------------- /data/evals/trex-test-causal-base.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/data/evals/trex-test-causal-base.json -------------------------------------------------------------------------------- /data/evals/trex-test-causal-sim10.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/data/evals/trex-test-causal-sim10.json -------------------------------------------------------------------------------- /data/evals/trex-test-masked-base.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/data/evals/trex-test-masked-base.json -------------------------------------------------------------------------------- /data/evals/trex-test-masked-sim10.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/data/evals/trex-test-masked-sim10.json -------------------------------------------------------------------------------- /data/evals/twitter-small-directed-test-causal-base.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/data/evals/twitter-small-directed-test-causal-base.json -------------------------------------------------------------------------------- /data/evals/twitter-small-directed-test-masked-base.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/data/evals/twitter-small-directed-test-masked-base.json -------------------------------------------------------------------------------- /data/evals/twitter-small-test-causal-base.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/data/evals/twitter-small-test-causal-base.json -------------------------------------------------------------------------------- /data/evals/twitter-small-test-causal-sim10.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/data/evals/twitter-small-test-causal-sim10.json -------------------------------------------------------------------------------- /data/evals/twitter-small-test-masked-base.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/data/evals/twitter-small-test-masked-base.json -------------------------------------------------------------------------------- /data/evals/twitter-small-test-masked-sim10.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/data/evals/twitter-small-test-masked-sim10.json -------------------------------------------------------------------------------- /data/link-prediction-eval.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/data/link-prediction-eval.csv -------------------------------------------------------------------------------- /data/pubmed-textinit/raw: -------------------------------------------------------------------------------- 1 | ../pubmed/raw -------------------------------------------------------------------------------- /data/pubmed/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/data/pubmed/.gitignore -------------------------------------------------------------------------------- /data/pubmed/prep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/data/pubmed/prep.py -------------------------------------------------------------------------------- /data/trex-textinit/raw: -------------------------------------------------------------------------------- 1 | ../trex/raw/ -------------------------------------------------------------------------------- /data/trex/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/data/trex/.gitignore -------------------------------------------------------------------------------- /data/trex/orig/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/data/trex/orig/README.md -------------------------------------------------------------------------------- /data/trex/prep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/data/trex/prep.py -------------------------------------------------------------------------------- /data/twitter-small-textinit/derived: -------------------------------------------------------------------------------- 1 | ../twitter-small/derived/ -------------------------------------------------------------------------------- /data/twitter-small-textinit/raw: -------------------------------------------------------------------------------- 1 | ../twitter-small/raw/ -------------------------------------------------------------------------------- /data/twitter-small/1-node-properties.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/data/twitter-small/1-node-properties.py -------------------------------------------------------------------------------- /notebooks/.ipynb_checkpoints/1a-datasets-checkpoint.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/notebooks/.ipynb_checkpoints/1a-datasets-checkpoint.ipynb -------------------------------------------------------------------------------- /notebooks/.ipynb_checkpoints/1b-save-splits-checkpoint.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/notebooks/.ipynb_checkpoints/1b-save-splits-checkpoint.ipynb -------------------------------------------------------------------------------- /notebooks/.ipynb_checkpoints/2a-eval-cross-checkpoint.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/notebooks/.ipynb_checkpoints/2a-eval-cross-checkpoint.ipynb -------------------------------------------------------------------------------- /notebooks/.ipynb_checkpoints/2a-eval-text-pubmed-checkpoint.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/notebooks/.ipynb_checkpoints/2a-eval-text-pubmed-checkpoint.ipynb -------------------------------------------------------------------------------- /notebooks/.ipynb_checkpoints/2b-eval-text-trex-checkpoint.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/notebooks/.ipynb_checkpoints/2b-eval-text-trex-checkpoint.ipynb -------------------------------------------------------------------------------- /notebooks/.ipynb_checkpoints/2c-eval-text-twitter-small-checkpoint.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/notebooks/.ipynb_checkpoints/2c-eval-text-twitter-small-checkpoint.ipynb -------------------------------------------------------------------------------- /notebooks/.ipynb_checkpoints/3a-eval-link-pred-checkpoint.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/notebooks/.ipynb_checkpoints/3a-eval-link-pred-checkpoint.ipynb -------------------------------------------------------------------------------- /notebooks/.ipynb_checkpoints/4a-eval-descriptive-pubmed-checkpoint.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/notebooks/.ipynb_checkpoints/4a-eval-descriptive-pubmed-checkpoint.ipynb -------------------------------------------------------------------------------- /notebooks/.ipynb_checkpoints/4b-eval-descriptive-trex-checkpoint.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/notebooks/.ipynb_checkpoints/4b-eval-descriptive-trex-checkpoint.ipynb -------------------------------------------------------------------------------- /notebooks/.ipynb_checkpoints/4c-eval-descriptive-twitter-small-checkpoint.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/notebooks/.ipynb_checkpoints/4c-eval-descriptive-twitter-small-checkpoint.ipynb -------------------------------------------------------------------------------- /notebooks/.ipynb_checkpoints/5a-eval-cross-checkpoint.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/notebooks/.ipynb_checkpoints/5a-eval-cross-checkpoint.ipynb -------------------------------------------------------------------------------- /notebooks/1a-datasets.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/notebooks/1a-datasets.ipynb -------------------------------------------------------------------------------- /notebooks/1b-save-splits.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/notebooks/1b-save-splits.ipynb -------------------------------------------------------------------------------- /notebooks/2a-eval-text-pubmed.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/notebooks/2a-eval-text-pubmed.ipynb -------------------------------------------------------------------------------- /notebooks/2b-eval-text-trex.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/notebooks/2b-eval-text-trex.ipynb -------------------------------------------------------------------------------- /notebooks/2c-eval-text-twitter-small.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/notebooks/2c-eval-text-twitter-small.ipynb -------------------------------------------------------------------------------- /notebooks/3a-eval-link-pred.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/notebooks/3a-eval-link-pred.ipynb -------------------------------------------------------------------------------- /notebooks/4a-eval-descriptive-pubmed.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/notebooks/4a-eval-descriptive-pubmed.ipynb -------------------------------------------------------------------------------- /notebooks/4b-eval-descriptive-trex.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/notebooks/4b-eval-descriptive-trex.ipynb -------------------------------------------------------------------------------- /notebooks/4c-eval-descriptive-twitter-small.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/notebooks/4c-eval-descriptive-twitter-small.ipynb -------------------------------------------------------------------------------- /notebooks/5a-eval-cross.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/notebooks/5a-eval-cross.ipynb -------------------------------------------------------------------------------- /notebooks/lr-tuning.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/notebooks/lr-tuning.ipynb -------------------------------------------------------------------------------- /notebooks/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/notebooks/utils.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/pyproject.toml -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/setup.py -------------------------------------------------------------------------------- /src/clip_graph/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/src/clip_graph/__init__.py -------------------------------------------------------------------------------- /src/clip_graph/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/src/clip_graph/data/__init__.py -------------------------------------------------------------------------------- /src/clip_graph/data/callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/src/clip_graph/data/callbacks.py -------------------------------------------------------------------------------- /src/clip_graph/data/datamodule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/src/clip_graph/data/datamodule.py -------------------------------------------------------------------------------- /src/clip_graph/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/src/clip_graph/data/dataset.py -------------------------------------------------------------------------------- /src/clip_graph/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/src/clip_graph/eval.py -------------------------------------------------------------------------------- /src/clip_graph/lit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/src/clip_graph/lit.py -------------------------------------------------------------------------------- /src/clip_graph/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/src/clip_graph/losses.py -------------------------------------------------------------------------------- /src/clip_graph/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/src/clip_graph/models.py -------------------------------------------------------------------------------- /src/clip_graph/optimizers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/src/clip_graph/optimizers.py -------------------------------------------------------------------------------- /src/clip_graph/scoring.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/src/clip_graph/scoring.py -------------------------------------------------------------------------------- /src/clip_graph/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wwbrannon/congrat/HEAD/src/clip_graph/utils.py --------------------------------------------------------------------------------