Skip to main content

Module cluster_encoder

Module cluster_encoder 

Source
Expand description

Cross-context clustering primitives (shared via anno-metrics).

Structs§

ClusterEmbedding
Cluster embedding: a fixed-size representation of a coreference cluster.
ClusterEncoderConfig
Configuration for cluster encoding.
ClusterMention
A mention within a cluster, represented by character offsets.
CosineMergeScorer
Simple cosine similarity scorer (CPU fallback).
CrossContextConfig
Configuration for cross-context resolution.
CrossContextResolver
Cross-context coreference resolution combining encoder and scorer.
HeuristicClusterEncoder
Simple heuristic cluster encoder using hashed character n-grams.
LocalCluster
A coreference cluster containing mentions from a single context.
MergeScorerConfig
Configuration for merge scoring.
MergedCluster
A merged cross-context cluster.

Enums§

PoolingStrategy
Pooling strategy for reducing mention embeddings to a cluster embedding.

Traits§

ClusterEncoder
Trait for encoding clusters into fixed-size embeddings.
MergeScorer
Trait for scoring cluster merge probability.