How to set number of thread

[davutpolat]

Hi,

i want to use mrmr for 1000 features and 10gb sized data, but it takes too long time, i got 32 core server ( can hire 64 core as well), but i see there is only 4 cores in use, does pymrmr run all cores? if so , how should i use it ?

Thanks in advance,

Regards,
Davut

[fbrundu]

Dear Davut,
You're right, it uses hard-coded 4 cores by employing openmp - see line 178 of mrmr.cpp:

pymrmr/pymrmr/mrmr.cpp

Line 178 in 1a5dbc8

#pragma omp parallel num_threads(4)

Right now, unfortunately, I have not the time to implement the definition of a parameter for the number of threads, but I'm open to PRs if you'd like to modify the code.
A more quick and dirty solution for your problem would be to fork the repo, modify that specific line with the number of cores you need, and install the repo again from your account using:

pip install --upgrade git+https://github.com/davutpolat/pymrmr

Best,
Francesco