graphembed 0.0.8

graph embedding
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31

//! do self loop augmentation as described in the [Nodesketch paper](https://dl.acm.org/doi/10.1145/3292500.3330951)

use sprs::{TriMatI, CsMatI};

use ndarray::{Array1};

// takes a matrix in a triplet form augment it by diagonal
pub(crate) fn diagonal_augmentation(graphtrip : &mut TriMatI<f64, usize>, weight : f64) -> CsMatI<f64, usize> {
    let shape = graphtrip.shape();
    log::debug!("diagonal_augmentation received shape {:?}", shape);
    assert_eq!(shape.0, shape.1);
    // determines max for each row
    let mut rowmax = Array1::<f64>::zeros(shape.0);
    let mut trip_iter = graphtrip.triplet_iter();
    while let Some(triplet) =  trip_iter.next() {
        if *triplet.0 > rowmax[triplet.1.0] {
            rowmax[triplet.1.0] = *triplet.0
        }
    }
    //
    for i in 0..shape.0 {
        // we do diagonal augmentation only for non isolated point
        if weight > 0. {
            // TODO could add rowmax[row]  if rowmax[row] > 0 ? Also beware of asymetric graph where node row can be only a target node!!
            graphtrip.add_triplet(i,i, weight);
//            log::trace!("diagonal_augmentation row : {}, value : {}", i, 1.);
        }
    }
    //
    graphtrip.to_csr()
} // end of diagonal_augmentation