`n_jobs` and `batch_size` should be set into the constructor instead. It would also be nice it you could include a docstring for both of them. It might indeed not be clear for everyone what is the meaning of `batch_size`.