CODEWITHSUNDEEP
scikit-learnparallel-processingneural-networkjoblib
strateeg32
strateeg32

Reputation: 168

Sklearn neural network with maximum number of cores available?

I want to use MLPRegressor from sklearn with all 12 cores available to me, however I do not see any option to select the amount of cores (such as with RandomForestClassifier which has the option with n_jobs).

Is there another way to make sure it uses all 12 cores? I vaguely heard about joblib, but how would I use it correctly?

Upvotes: 2

Views: 744

Answers (1)

eschibli
eschibli

Reputation: 872

MLPRegressor does not contain any multithreading per se, though the matrix operations will be vectorized and parallelized via numpy.

You may be able to get better performance by varying your batch size, but if performance is critical you should use a deep learning library like Tensorflow.

Upvotes: 1

Related Questions