search

Gorilla-Lab-SCUT / GPNet

28 stars 5 forks source link

matrix_k_min function #3

Closed antoalli closed 2 years ago

antoalli commented 3 years ago

Hi! I was wondering what is actually doing the 'matrix_k_min' function, for example here https://github.com/CZ-Wu/GPNet/blob/59419f4505ca0732316a7d993bfe065547892847/tools/proposal.py#L396. What is actual being performed in the 389-398' for loop and what 'matrix_k_min' is accomplishing?

Thank you very much

CZ-Wu commented 3 years ago

It looks for the top k minimum values along each row of the matrix and return the indices. The loop there is to avoid out of memory. This function is for calculating the fixed number of neighbors of the contacts.