distortions/lib/reshape.js

Helpers for visualizing distortion in nonlinear dimensionality reduction.

// flatten the neighborhoods into the link data
//
// example:
// [{1: [2, 3, 4], 2: [1, 3, 4]}] --> [[1, 2], [1, 3], ..., [2, 4]]
export function flatten_edges(N, dataset, mappingObj) {
    let link_data = []
    for (let Ni in N) {
        for (let Nj in N[Ni]) {
            link_data.push({
                "_id": `${Ni}_${N[Ni][Nj]}`,
                x1: dataset[Ni][mappingObj.x], y1: dataset[Ni][mappingObj.y],
                x2: dataset[N[Ni][Nj]][mappingObj.x], y2: dataset[N[Ni][Nj]][mappingObj.y],
                "_id_source": Ni,
                "_id_target": N[Ni][Nj],
            })
        }
    }
    return link_data
}