text-graph-generalization

Code for our paper Compositional Generalization with Grounded Language Models
git clone https://esimon.eu/repos/text-graph-generalization.git
Log | Files | Refs | README | LICENSE

ModeNameSize
-rw-r--r--.gitignore162L
-rw-r--r--LICENSE674L
-rw-r--r--README.md93L
-rw-r--r--bidirectional.py834L
-rw-r--r--calculate_graph_statistics.py28L
-rw-r--r--dataset.py148L
-rw-r--r--dataset_maker.py259L
-rw-r--r--device.py2L
-rw-r--r--download_data.md65L
-rwxr-xr-xdownload_data.sh5L
-rw-r--r--graph_sampler.py294L
-rw-r--r--model.py116L
-rw-r--r--requirements.txt12L
-rw-r--r--results/results.json182L
-rw-r--r--static/2_hop_annotations.json1L
-rw-r--r--static/3_hop_annotations.json586L
-rw-r--r--static/4_hop_annotations.json1784L
-rw-r--r--static/annotation_map.json2564L
-rw-r--r--static/edge_type_dict.json1L
-rw-r--r--train.py221L
-rw-r--r--utils.py27L
-rw-r--r--write_results.py49L