MR-H2SAGE intermediate files: graph and model results for "Not all edges are equal: separating edge types in graph neural networks for lithology prediction from borehole logs"

Abstract

Intermediate files of the MR-H2SAGE workflow (https://gitlab.com/frederikke-storm-phd/mr-h2sage) to reproduce every figure and table of the paper with notebook 05: graph_data_structure.pkl: the PyTorch Geometric graph of 63 Danish boreholes with processed borehole logs, lithology classes and depth, spatial and feature-similarity edges (written by notebook 02); 03_model_training/: results of the five models on the full graph, one subfolder per model (written by notebook 03); 04_edge_ablation/: results of MR-H2SAGE and GraphSAGE retrained on each single edge type, one subfolder per model and edge type (written by notebook 04). Unzip in the repository folder; the paths in notebook 05 then work as they are. The pickle requires the package versions in the repository's requirements.txt. The files are derived from borehole geophysical logs in the GERDA database and lithological descriptions in the Jupiter database, both hosted by the Geological Survey of Denmark and Greenland (GEUS). DGU numbers of the boreholes are listed in the repository (BoreholeLogs_data/fold_assignments.csv).