pythonscikit-learn

how to use sklearn when target variable is a proportion


There are standard ways of predicting proportions such as logistic regression (without thresholding) and beta regression. There have already been discussions about this:

http://scikit-learn-general.narkive.com/4dSCktaM/using-logistic-regression-on-a-continuous-target-variable

http://scikit-learn-general.narkive.com/lLVQGzyl/beta-regression

I cannot tell if there exists a work-around within the sklearn framework.


Solution

  • There exists a workaround, but it is not intrinsically within the sklearn framework.

    If you have a proportional target variable (value range 0-1) you run into two basic difficulties with scikit-learn:

    There are different ways to mathematically formulate logistic regression. One of them is the generalized linear model, which basically defines the logistic regression as a normal linear regression on logit-transformed probabilities. Normally, this approach requires sophisticated mathematical optimization because the probabilities are unknown and need to be estimated along with the regression coefficients.

    In your case, however, the probabilities are known. This means you can simply transform them with y = log(p / (1 - p)). Now they cover the full range from -oo to oo and can serve as the target variable for a LinearRegression model [*]. Of course, the model output then needs to be transformed again to result in probabilities p = 1 / (exp(-y) + 1).

    import numpy as np
    from sklearn.linear_model import LinearRegression
    
    
    class LogitRegression(LinearRegression):
    
        def fit(self, x, p):
            p = np.asarray(p)
            y = np.log(p / (1 - p))
            return super().fit(x, y)
    
        def predict(self, x):
            y = super().predict(x)
            return 1 / (np.exp(-y) + 1)
    
    
    if __name__ == '__main__':
        # generate example data
        np.random.seed(42)
        n = 100
        x = np.random.randn(n).reshape(-1, 1)
        noise = 0.1 * np.random.randn(n).reshape(-1, 1)
        p = np.tanh(x + noise) / 2 + 0.5
    
        model = LogitRegression()
        model.fit(x, p)
    
        print(model.predict([[-10], [0.0], [1]]))
        # [[  2.06115362e-09]
        #  [  5.00000000e-01]
        #  [  8.80797078e-01]]
    

    [*] You could in fact plug in any linear regression model which can make the method more powerful, but then it no longer is exactly equivalent to logistic regression.