dask / dask-glm

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Compute_stepsize function in proximal_grad

mrocklin opened this issue · comments

Currently the stepsize computation in proximal_grad is still embedded within the code, which makes it hard to manage it with Dask. If the proximal_grad method is useful then it would be nice to pull this function out.

cc @moody-marlin

So this is where the regularization stuff gets hairy; proximal_grad is an algorithm for regularized logistic problems, which means the function evaluations within compute_stepsize need to be a modified loglike function; this in turn makes jitting compute_stepsize more complicated because it now needs a function argument.