Панды не работают с линейной регрессией - PullRequest
0 голосов
/ 07 июля 2019

Я пытаюсь выполнить регрессию некоторых данных из фрейма данных, но я продолжаю получать эту странную ошибку формы.Любая идея, что не так?

import pandas as pd
import io
import requests
import statsmodels.api as sm

# Read in a dataset 
url="https://raw.githubusercontent.com/jldbc/coffee-quality-database/master/data/arabica_data_cleaned.csv"
s=requests.get(url).content
df=pd.read_csv(io.StringIO(s.decode('utf-8')))

# Select feature columns 
X = df[['Body', 'Clean.Cup']]

# Select dv column
y = df['Cupper.Points']

# make model
mod = sm.OLS(X, y).fit()

Я получаю эту ошибку: формы (1311,2) и (1311,2) не выровнены: 2 (dim 1)! = 1311 (dim 0)

Ответы [ 2 ]

0 голосов
/ 07 июля 2019

Неправильный порядок y и X.

sm.OLS(y,X)
0 голосов
/ 07 июля 2019

У вас есть X и y термины в неправильном порядке в вашей sm.OLS команде:

import pandas as pd
import io
import requests
import statsmodels.api as sm

# Read in a dataset 
url="https://raw.githubusercontent.com/jldbc/coffee-quality-database/master/data/arabica_data_cleaned.csv"
s=requests.get(url).content
df=pd.read_csv(io.StringIO(s.decode('utf-8')))

# Select feature columns 
X = df[['Body', 'Clean.Cup']]

# Select dv column
y = df['Cupper.Points']

# make model
mod = sm.OLS(y, X).fit()

mod.summary()

запускается и возвращает

<class 'statsmodels.iolib.summary.Summary'>
"""
                            OLS Regression Results                            
==============================================================================
Dep. Variable:          Cupper.Points   R-squared:                       0.998
Model:                            OLS   Adj. R-squared:                  0.998
Method:                 Least Squares   F-statistic:                 3.145e+05
Date:                Sat, 06 Jul 2019   Prob (F-statistic):               0.00
Time:                        19:42:59   Log-Likelihood:                -454.94
No. Observations:                1311   AIC:                             913.9
Df Residuals:                    1309   BIC:                             924.2
Df Model:                           2                                         
Covariance Type:            nonrobust                                         
==============================================================================
                 coef    std err          t      P>|t|      [0.025      0.975]
------------------------------------------------------------------------------
Body           0.8464      0.016     53.188      0.000       0.815       0.878
Clean.Cup      0.1154      0.012      9.502      0.000       0.092       0.139
==============================================================================
Omnibus:                      537.879   Durbin-Watson:                   1.710
Prob(Omnibus):                  0.000   Jarque-Bera (JB):            30220.027
Skew:                           1.094   Prob(JB):                         0.00
Kurtosis:                      26.419   Cond. No.                         26.2
==============================================================================

Warnings:
[1] Standard Errors assume that the covariance matrix of the errors is correctly specified.
"""
...