def __init__(self, n_jobs: int=-1, parallel_backend: str='joblib') -> None:
"""
Construct OptimalK to use n_jobs (multiprocessing using joblib, multiprocessing, or single core.
:param n_jobs - int: Number of CPU cores to use. Use all cores if n_jobs == -1
"""
self.parallel_backend = parallel_backend if parallel_backend in ['joblib', 'multiprocessing'] else None
self.n_jobs = n_jobs if 1 <= n_jobs <= cpu_count() else cpu_count() # type: int
self.n_jobs = 1 if parallel_backend is None else self.n_jobs
评论列表
文章目录