Wikidata Thematic Subgraph Selection

<p><strong>Wikidata Thematic Subgraph Selection</strong></p> <p>These datasets have been designed to train and evaluate algorithms to select thematic subgraphs of interest in a large knowledge graph from seed entities of interest. Specifically, we consider Wikidata. Given a set of seed QIDs of interest, a graph expansion is performed following P31, P279, and (-)P279 edges. Traversed classes that thematically deviates from seed QIDs of interest should be pruned. Datasets thus consist of classes reached from seed QIDs that are labeled as &quot;to prune&quot; or &quot;to keep&quot;.</p> <p><strong>Available datasets</strong></p> <table> <tbody><tr> <th>Dataset</th> <th># Seed QIDs</th> <th># Labeled decisions</th> <th># Prune decisions</th> <th>Min prune depth</th> <th>Max prune depth</th> <th># Keep decisions</th> <th>Min keep depth</th> <th>Max keep depth</th> <th># Reached nodes up</th> <th># Reached nodes down</th> </tr> </tbody><tbody> <tr> <td><a href="data/dataset1">dataset1</a></td> <td>455</td> <td>5233</td> <td>3464</td> <td>1</td> <td>4</td> <td>1769</td> <td>1</td> <td>4</td> <td>1507</td> <td>2593609</td> </tr> <tr> <td><a href="data/dataset2">dataset2</a></td> <td>105</td> <td>982</td> <td>388</td> <td>1</td> <td>2</td> <td>594</td> <td>1</td> <td>3</td> <td>1159</td> <td>1247385</td> </tr> </tbody> </table> <p>Each dataset folder contains</p> <ul> <li><code>datasetX.csv</code>: a CSV file containing one seed QID per line (not the complete URL, just the QID). This CSV file has no header.</li> <li><code>datasetX_labels.csv</code>: a CSV file containing one seed QID per line and its label (not the complete URL, just the QID)</li> <li><code>datasetX_gold_decisions.csv</code>: a CSV file with seed QIDs, reached QIDs, and the labeled decision (1: keep, 0: prune)</li> <li><code>datasetX_Y_folds.pkl</code>: folds to train and test models based on the labeled decisions</li> </ul> <p><code>dataset1-2</code> consists of using <code>dataset1</code> for training and <code>dataset2</code> for testing.</p> <p><strong>License</strong></p> <p>Datasets are available under the <a href="https://creativecommons.org/licenses/by-nc/4.0/">CC BY-NC</a> license.</p>