python - 值错误 : endog must be in the unit interval

Question

在使用 statsmodels 时，我收到了这个奇怪的错误:ValueError: endog must be in the unit interval. 谁能给我更多关于这个错误的信息？谷歌没有帮助。

产生错误的代码:

"""
Multiple regression with dummy variables. 
"""

import pandas as pd
import statsmodels.api as sm
import pylab as pl
import numpy as np

df = pd.read_csv('cost_data.csv')
df.columns = ['Cost', 'R(t)', 'Day of Week']
dummy_ranks = pd.get_dummies(df['Day of Week'], prefix='days')
cols_to_keep = ['Cost', 'R(t)']
data = df[cols_to_keep].join(dummy_ranks.ix[:,'days_2':])
data['intercept'] = 1.0

print(data)

train_cols = data.columns[1:]
logit = sm.Logit(data['Cost'], data[train_cols])

result = logit.fit()

print(result.summary())

和回溯:

Traceback (most recent call last):
  File "multiple_regression_dummy.py", line 20, in <module>
    logit = sm.Logit(data['Cost'], data[train_cols])
  File "/Library/Frameworks/Python.framework/Versions/3.4/lib/python3.4/site-packages/statsmodels/discrete/discrete_model.py", line 404, in __init__
    raise ValueError("endog must be in the unit interval.")
ValueError: endog must be in the unit interval.

Answer 1

当我的目标列的值大于 1 时出现此错误。 确保您的目标列介于 0 和 1 之间(这是逻辑回归所必需的)，然后重试。 例如，如果目标列的值为 1-5，则将 4 和 5 设为正类，将 1、2、3 设为负类。希望这会有所帮助。