Struct CooccurrenceFeatures 

pub struct CooccurrenceFeatures {
    pub entity_key: String,
    pub cooccurring_entities: Vec<String>,
    pub cooccurrence_counts: HashMap<String, usize>,
    pub total_cooccurrences: usize,
    pub unique_cooccurrences: usize,
    pub cooccurring_types: HashMap<String, Vec<String>>,
}

Co-occurrence features for an entity.

Fields

entity_key: String

The entity (normalized key).

cooccurring_entities: Vec<String>

Entities that co-occur within the window.

cooccurrence_counts: HashMap<String, usize>

Co-occurrence counts per entity.

total_cooccurrences: usize

Total co-occurrence count (sum of all).

unique_cooccurrences: usize

Unique co-occurring entity count.

cooccurring_types: HashMap<String, Vec<String>>

Entity types of co-occurring entities.

Implementations

impl CooccurrenceFeatures

pub fn new(entity_key: String) -> Self

Create new co-occurrence features for an entity.

pub fn add_cooccurrence(&mut self, other_key: &str, other_type: Option<&str>)

Add a co-occurring entity.

pub fn finalize(&mut self)

Finalize the features (dedupe, sort, count).

pub fn top_k(&self, k: usize) -> Vec<(&str, usize)>

Get top-k co-occurring entities by count.

Trait Implementations

impl Clone for CooccurrenceFeatures

fn clone(&self) -> CooccurrenceFeatures

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for CooccurrenceFeatures

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.