VHRanger / nodevectors

Fastest network node embeddings in the west

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

ProNE multithread

ashkspark opened this issue · comments

Just want to know if ProNE is multithreaded? Is there a way to control the number of threads like the implemented Node2Vec?

Hey,

The ProNE algorithm is inherently sequential. Right now it mainly leverages scipy and numpy:

https://github.com/VHRanger/nodevectors/blob/master/nodevectors/prone.py#L155

Some parts could be optimized (the sparse matrix multiplications could be rewritten in optimized numba or C++) which could leverage a little bit of parallelism, but nothing like the node2vec which is an inherently parallel algorithm (generating random walks in embarassingly parallel)