Struct ReferenceGraph 

pub struct ReferenceGraph {
    pub nodes: Vec<String>,
    pub edges: Vec<(String, String, ReferenceType, f64)>,
}

Reference graph for tracking relationships between documents.

Used by tier for hierarchical clustering of entity relationships.

Fields

nodes: Vec<String>

Nodes: document IDs

edges: Vec<(String, String, ReferenceType, f64)>

Edges: (source_doc, target_doc, reference_type, weight)

Implementations

impl ReferenceGraph

pub fn new() -> Self

Create a new reference graph.

pub fn add_document(&mut self, doc_id: &str)

Add a document node.

pub fn add_reference( &mut self, source_doc: &str, target_doc: &str, ref_type: ReferenceType, weight: f64, )

Add a reference edge between documents.

pub fn get_references(&self, doc_id: &str) -> Vec<(&str, &ReferenceType)>

Get all documents referenced by a given document.

pub fn get_referrers(&self, doc_id: &str) -> Vec<(&str, &ReferenceType)>

Get all documents that reference a given document.

pub fn get_depth(&self, doc_id: &str) -> usize

Get reference depth (minimum hops from root documents).

Root documents are those with no incoming references.

pub fn to_graph_edges(&self) -> Vec<(usize, usize, f64)>

Convert to a format suitable for tier clustering.

Trait Implementations

impl Clone for ReferenceGraph

fn clone(&self) -> ReferenceGraph

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ReferenceGraph

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

Formats the value using the given formatter.

impl Default for ReferenceGraph

fn default() -> ReferenceGraph

Returns the “default value” for a type.

impl<'de> Deserialize<'de> for ReferenceGraph

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for ReferenceGraph

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.