What does the delayed() function do (when used with joblib in Python)
we need a loop to test a list of different model configurations. This is the main function that drives the grid search process and will call the score_model() function for each model configuration. We can dramatically speed up the grid search process by evaluating model configurations in parallel. One way to do that is to use the Joblib library . We can define a Parallel object with the number of cores to use and set it to the number of scores detected in your hardware.
define executor
executor = Parallel(n_jobs=cpu_count(), backend= 'multiprocessing' )
then create a list of tasks to execute in parallel, which will be one call to the score model() function for each model configuration we have.
suppose def score_model(data, n_test, cfg):
........................
define list of tasks
tasks = (delayed(score_model)(data, n_test, cfg) for cfg in cfg_list)
we can use the Parallel object to execute the list of tasks in parallel.
scores = executor(tasks)
Perhaps things become clearer if we look at what would happen if instead we simply wrote
Parallel(n_jobs=8)(getHog(i) for i in allImages)
which, in this context, could be expressed more naturally as:
- Create a
Parallel
instance withn_jobs=8
- create the list
[getHog(i) for i in allImages]
- pass that list to the
Parallel
instance
What's the problem? By the time the list gets passed to the Parallel
object, all getHog(i)
calls have already returned - so there's nothing left to execute in Parallel! All the work was already done in the main thread, sequentially.
What we actually want is to tell Python what functions we want to call with what arguments, without actually calling them - in other words, we want to delay the execution.
This is what delayed
conveniently allows us to do, with clear syntax. If we want to tell Python that we'd like to call foo(2, g=3)
sometime later, we can simply write delayed(foo)(2, g=3)
. Returned is the tuple (foo, [2], {g: 3})
, containing:
- a reference to the function we want to call, e.g.
foo
- all arguments (short "args") without a keyword, e.g.t
2
- all keyword arguments (short "kwargs"), e.g.
g=3
So, by writing Parallel(n_jobs=8)(delayed(getHog)(i) for i in allImages)
, instead of the above sequence, now the following happens:
A
Parallel
instance withn_jobs=8
gets createdThe list
[delayed(getHog)(i) for i in allImages]
gets created, evaluating to
[(getHog, [img1], {}), (getHog, [img2], {}), ... ]
That list is passed to the
Parallel
instanceThe
Parallel
instance creates 8 threads and distributes the tuples from the list to themFinally, each of those threads starts executing the tuples, i.e., they call the first element with the second and the third elements unpacked as arguments
tup[0](*tup[1], **tup[2])
, turning the tuple back into the call we actually intended to do,getHog(img2)
.