ENH: Add inertially accelerated proximal cd

solvers/apcd.py

@@ -0,0 +1,37 @@
+# from benchopt import BaseSolver, safe_import_context
+from benchopt import BaseSolver
+
+# with safe_import_context() as ctx:
+
+
+class Solver(BaseSolver):
+    """Accelerated proximal coordinate descent by Fercoq and Bianchi."""
+    name = 'apcd'
+    stop_strategy = 'iteration'
+
+    install_cmd = "conda"
+    requirements = [
+        "pip:git+https://bitbucket.org/mathurinm/"
+        "cd_solver.git@module_structure"]
+
+    def set_objective(self, X, y, lmbd, fit_intercept):
+        self.X, self.y, self.lmbd = X, y, lmbd
+        self.fit_intercept = fit_intercept
+
+    def run(self, n_iter):
+        n_samples, n_features = self.X.shape
+        f = ["square"] * n_samples
+        N = n_features
+        Af = self.X
+        bf = self.y
+        cf = [0.5] * n_samples
+        g = ["abs"] * n_features
+        cg = [self.lmbd] * n_features
+        import cd_solver
+        pb = cd_solver.Problem(N=N, f=f, Af=Af, bf=bf, cf=cf, g=g, cg=cg)
+        cd_solver.cd_solver_.coordinate_descent(
+            pb, max_iter=n_iter)
+        self.w = pb.sol
+
+    def get_result(self):
+        return self.w