Обучение моделям машинного обучения Scikit-learn с использованием нескольких процессоров - PullRequest
0 голосов
/ 03 сентября 2018

Я хочу сократить время обучения своих моделей, используя высокопроизводительный экземпляр EC2. Поэтому я попробовал экземпляр c5.18xlarge с 2 процессорами и запустил несколько моделей с параметром n_jobs=-1, но заметил, что использовался только один процессор:

enter image description here

Можно ли как-нибудь заставить Scikit-научиться использовать все процессоры?

1 Ответ

0 голосов
/ 24 сентября 2018

Попробуйте добавить:

import multiprocessing
multiprocessing.set_start_method('forkserver')

вверху вашего кода, прежде чем запускать или импортировать что-либо. Это хорошо известная проблема с многопроцессорностью в python.

...