pymap_elites icon indicating copy to clipboard operation
pymap_elites copied to clipboard

Published 20 hours ago • verified publisher icon resibots

n_jobs deprecated in sklearn Kmeans

Open Aneoshun opened this issue 2 years ago • 1 comments

Hi,

Running the cvt_rastrigin example (but I guess that's valid for any example) raises: TypeError: KMeans.__init__() got an unexpected keyword argument 'n_jobs'

Removing this argument here solves it: https://github.com/resibots/pymap_elites/blob/master/map_elites/common.py#L194

However, I cannot find in the doc what this argument what used for, so I leave this to you @jbmouret to decide if it is safe to remove it.

Best,

Aneoshun avatar Sep 20 '23 08:09 Aneoshun

It was supposed to enable parallelization of Kmeans, but apparently they removed the argument. Please make a PR without it :)

jbmouret avatar Sep 20 '23 08:09 jbmouret