sklearn.utils.register_parallel_backend?

sklearn.utils.register_parallel_backend(name, factory, make_default=False)

源碼

注冊一個新的并行后端工廠。

然后,可以通過將其名稱作為后端參數傳遞給Parallel類來選擇新的后端。 此外,可以通過設置make_default = True全局覆蓋默認后端。

工廠可以是任何不帶參數的可調用對象,并且返回ParallelBackendBase的實例。

警告:此功能是實驗性的,在Joblib的將來版本中可能會更改。

版本0.10中的新功能。