Преобразование остаточной стоимости Numpy Lstsq в R ^ 2 - PullRequest
10 голосов
/ 16 июня 2010

Я выполняю регрессию по методу наименьших квадратов, как показано ниже (одномерный). Я хотел бы выразить значимость результата в терминах R ^ 2. Numpy возвращает значение немасштабированного остатка, что является разумным способом нормализации этого.

field_clean,back_clean = rid_zeros(backscatter,field_data)
num_vals = len(field_clean)
x = field_clean[:,row:row+1]
y = 10*log10(back_clean)

A = hstack([x, ones((num_vals,1))])
soln = lstsq(A, y )
m, c =  soln [0]
residues = soln [1]

print residues

1 Ответ

18 голосов
/ 17 июня 2010

См. http://en.wikipedia.org/wiki/Coefficient_of_determination

Ваше значение R2 =

1 - residual / sum((y - y.mean())**2) 

, что эквивалентно

1 - residual / (n * y.var())

В качестве примера:

import numpy as np

# Make some data...
n = 10
x = np.arange(n)
y = 3 * x + 5 + np.random.random(n)

# Note that polyfit is an easier way to do this...
# It would just be "model, resid = np.polyfit(x,y,1,full=True)[:2]" 
A = np.vstack((x, np.ones(n))).T
model, resid = np.linalg.lstsq(A, y)[:2]

r2 = 1 - resid / (y.size * y.var())
print r2
...