Struct StochasticBlockModel 

pub struct StochasticBlockModel {
    pub config: StochasticBlockModelConfig,
}

The Stochastic Block Model.

Fields

config: StochasticBlockModelConfig

Configuration.

Implementations

impl StochasticBlockModel

pub fn new(config: StochasticBlockModelConfig) -> Self

Create a new SBM with the given configuration.

pub fn fit(&self, graph: &AdjacencyGraph) -> Result<SBMResult>

Fit the SBM to the observed adjacency graph.

If config.num_blocks is set, fits with that K. Otherwise, tries each K in config.k_range and picks the one with the best ICL score.

pub fn predict( &self, graph: &AdjacencyGraph, b_matrix: &[f64], k: usize, ) -> Result<Vec<usize>>

Predict block assignments for a given graph using a fitted model.

This re-runs the E-step with the given B matrix to assign labels.

pub fn generate( n: usize, k: usize, b_matrix: &[f64], block_sizes: &[usize], seed: u64, ) -> Result<(AdjacencyGraph, Vec<usize>)>

Generate a random graph from SBM parameters.

• n: number of nodes
• k: number of blocks
• b_matrix: K x K probability matrix (row-major)
• block_sizes: sizes of each block (must sum to n)

Trait Implementations

impl Clone for StochasticBlockModel

fn clone(&self) -> StochasticBlockModel

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for StochasticBlockModel

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for StochasticBlockModel

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

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for StochasticBlockModel

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

where __S: Serializer,

Serialize this value into the given Serde serializer.