lbfgs_nnls.py

# (C) Mathieu Blondel 2012

import numpy as np
from scipy.optimize import fmin_l_bfgs_b

from sklearn.base import BaseEstimator, RegressorMixin
from sklearn.utils.extmath import safe_sparse_dot


class LbfgsNNLS(BaseEstimator, RegressorMixin):

    def __init__(self, tol=1e-3, callback=None, alpha=1.0):
        self.tol = tol
        self.callback = callback
        self.alpha = alpha

    def fit(self, X, y):
        n_features = X.shape[1]

        def f(w, *args):
            return (np.sum((safe_sparse_dot(X, w) - y) ** 2) +
                    self.alpha * np.sum(w))

        def fprime(w, *args):
            if self.callback is not None:
                self.coef_ = w
                self.callback(self)
            return self.alpha + 2 * safe_sparse_dot(X.T,
                                                    safe_sparse_dot(X, w) - y)

        coef0 = np.zeros(n_features, dtype=np.float64)
        w, f, d = fmin_l_bfgs_b(f, x0=coef0, fprime=fprime, pgtol=self.tol,
                                bounds=[(0, None)] * n_features)
        self.coef_ = w

        return self

    def n_nonzero(self, percentage=False):
        nz = np.sum(self.coef_ != 0)

        if percentage:
            nz /= float(self.coef_.shape[0])

        return nz

    def predict(self, X):
        return safe_sparse_dot(X, self.coef_)