CoolFace
Datasetpublic

brandburner/startrektng-s06-narrative-kg

Star Trek: The Next Generation - Narrative Knowledge Graph A rich narrative knowledge graph extracted from Star Trek: The Next Generation screenplays using the Fabula pipeline. Contains characters, locations, objects, organizations, events, themes, and conflict arcs with full participation semantics and Graph Gravity importance tiers. Dataset Overview Metric Value Source database startrektng.s06 Type Season database Episodes 26 Total nodes 7,541… See the full description on the dataset page: https://huggingface.co/datasets/brandburner/startrektng-s06-narrative-kg.

sourceHugging Facecc-by-sa-4.0updated 15d agoView on Hugging Face
0likes58downloads
Dataset Card

Star Trek: The Next Generation - Narrative Knowledge Graph

A rich narrative knowledge graph extracted from Star Trek: The Next Generation screenplays using the Fabula pipeline. Contains characters, locations, objects, organizations, events, themes, and conflict arcs with full participation semantics and Graph Gravity importance tiers.

Dataset Overview

MetricValue
Source databasestartrektng.s06
TypeSeason database
Episodes26
Total nodes7,541
Total edges25,190
Schema version1.2.1
Exported2026-09-12

Entity Breakdown

TypeCount
Act133
Agent447
ConflictArc54
Episode26
Event1,323
Location468
Object1,462
Organization128
PlotBeat2,605
SceneBoundary824
Theme59
Writer12

Graph Gravity Tiers

TierCountDescription
anchor35Main characters / key locations
planet152Recurring entities
asteroid2,318Minor / one-off entities

Relationship Types

AFFILIATED_WITH, BELONGS_TO_EPISODE, CALLBACK, CAUSAL, CHARACTER_CONTINUITY, CONTAINS_ACT, CONTAINS_BEAT, CONTAINS_SCENE, CREDITED_ON, EMOTIONAL_ECHO, ESCALATION, EXEMPLIFIES_THEME, FORESHADOWING, INVOLVED_IN_ARC, INVOLVED_WITH, IN_EVENT, NARRATIVELY_FOLLOWS, OCCURS_IN, PARTICIPATED_AS, PART_OF ... and 7 more

Related Datasets

This is a single-season dataset containing entities and events as extracted from Season 6 screenplays.

Note: The megagraph is not a simple union of season datasets. Cross-season entities are reconciled through a Global Entity Registry (GER), receiving new canonical UUIDs and distilled descriptions. Graph Gravity tiers are recalculated across all episodes. Use individual season datasets for single-season analysis; use the megagraph for cross-season analysis.

Files

FileDescription
nodes.parquetAll graph nodes with properties
edges.parquetAll relationships with properties
positions.parquet3D layout coordinates for visualization
meta.jsonDataset metadata and entity counts

Schema

Nodes (nodes.parquet)

ColumnTypeDescription
node_idstringUnique node identifier (UUID)
primary_labelstringNode type (Agent, Location, Event, etc.)
namestringDisplay name
descriptionstringFoundational description
tierstring (nullable)Graph Gravity tier: anchor / planet / asteroid
episode_countint (nullable)Number of distinct episodes entity appears in
first_episode_seqint (nullable)First appearance episode
last_episode_seqint (nullable)Last appearance episode
properties_jsonstringFull node properties as JSON

Edges (edges.parquet)

ColumnTypeDescription
source_node_idstringSource node UUID
target_node_idstringTarget node UUID
relationship_typestringRelationship type (e.g., PARTICIPATED_AS)
properties_jsonstringEdge properties as JSON

All relationship properties are carried verbatim inside properties_json. Notably, PARTICIPATED_AS edges may carry incarnation_identifier (the extractor's free-text label for the identity the character appears under) and, where the persona normalisation pass has run, persona — a controlled value reused verbatim across episodes, absent when the character appears as themselves (schema v1.2.1).

Positions (positions.parquet)

ColumnTypeDescription
node_idstringNode UUID
x, y, zfloat3D coordinates
sizefloatNode size (Graph Gravity weighted)
r, g, bintRGB color by entity type
communityintLouvain community index (seeded, deterministic)
tierstring (nullable)Graph Gravity tier
Layout method (schema ≥ 1.2.0): Coordinates are derived from the entities' semantic text embeddings (UMAP with a fixed seed and PCA initialization), so narratively similar entities sit near each other. Non-embedded nodes (events, scenes, episodes, etc.) are placed at the weighted barycenter of their narrative neighbours. The layout is deterministic: re-exporting an unchanged graph reproduces identical coordinates, and lightly-changed graphs keep comparable layouts. Not comparable with positions published under schema ≤ 1.1.0, which used a non-deterministic node2vec structural embedding. See meta.json → positions for the exact method and coverage stats.

Usage

python
from datasets import load_dataset
import pandas as pd

# Load from HuggingFace
ds = load_dataset("brandburner/startrektng-s06-narrative-kg")

# Or load parquet directly
nodes = pd.read_parquet("nodes.parquet")
edges = pd.read_parquet("edges.parquet")

# Filter to anchor characters
anchors = nodes[(nodes['primary_label'] == 'Agent') & (nodes['tier'] == 'anchor')]

# Build a NetworkX graph
import networkx as nx
G = nx.DiGraph()
for _, n in nodes.iterrows():
    G.add_node(n['node_id'], label=n['primary_label'], name=n['name'])
for _, e in edges.iterrows():
    G.add_edge(e['source_node_id'], e['target_node_id'], type=e['relationship_type'])

Citation

bibtex
@misc{fabula_startrektng_s06,
  title = {Star Trek: The Next Generation Narrative Knowledge Graph},
  author = {Fabula Pipeline},
  year = {2026},
  publisher = {HuggingFace},
  howpublished = {\url{https://huggingface.co/datasets/brandburner/startrektng-s06-narrative-kg}}
}

License

CC BY-SA 4.0