Patsy 的 dmatrices 无法读取我的公式

Patsy's dmatrices cannot read my formula

我有一个函数LogReg,如下所示:(以justmarkham's code为灵感)

def LogReg(self):
      formulA = "class ~"
      print self.frame #dataframe used
      print self.columnNames[:-1]
      for a in self.columnNames[:-1]:
         formulA += " {0} +".format(a)
      formula = formulA[:-2] #there is always a \n behind, we don't want that
      print "formula = " + formula
      Y,X = dmatrices(formula, self.frame, return_type="dataframe")
      Y = np.ravel(Y) #flatten Y to a 1D list
      model = LogisticRegression() #from sklearn.linear_model
      model = model.fit(X, Y)
      print model.score(X, Y)

结果如下:

         a0 a1  a2  a3 class
picture1  1  2   3  67     1
picture2  6  7  45  61     3
picture3  8  7   6   5     2
picture4  1  2   4   3     0
['a0', 'a1', 'a2', 'a3']
formula = class ~ a0 + a1 + a2 + a3
Traceback (most recent call last):
  File "classification.py", line 80, in <module>
    c.LogReg()
  File "classification.py", line 61, in LogReg
    Y,X = dmatrices(formula, self.frame, return_type="dataframe")
  File "/<path>/python2.7/site-packages/patsy/highlevel.py", line 297, in dmatrices
    NA_action, return_type)
  File "/<path>/python2.7/site-packages/patsy/highlevel.py", line 152, in _do_highlevel_design
    NA_action)
  File "/<path>/python2.7/site-packages/patsy/highlevel.py", line 57, in _try_incr_builders
    NA_action)
  File "/<path>/python2.7/site-packages/patsy/build.py", line 660, in design_matrix_builders
    NA_action)
  File "/<path>/python2.7/site-packages/patsy/build.py", line 424, in _examine_factor_types
    value = factor.eval(factor_states[factor], data)
  File "/<path>/python2.7/site-packages/patsy/eval.py", line 485, in eval
    return self._eval(memorize_state["eval_code"], memorize_state, data)
  File "/<path>/python2.7/site-packages/patsy/eval.py", line 468, in _eval
    code, inner_namespace=inner_namespace)
  File "/<path>/python2.7/site-packages/patsy/compat.py", line 117, in call_and_wrap_exc
    return f(*args, **kwargs)
  File "/<path>/python2.7/site-packages/patsy/eval.py", line 125, in eval
    code = compile(expr, source_name, "eval", self.flags, False)
  File "<string>", line 1
    class
        ^
SyntaxError: unexpected EOF while parsing

我看不出这里出了什么问题,因为据我所知该字符串不包含 EOF 字符,Python 代码似乎也没有错误。因此,问题:哪里出了问题(最好是:,以及如何解决)?

P.S.: 所用软件均为2015年4月9日可用的最新稳定包。

好吧,这很快。通过提问,我突然在代码中有颜色标记,通知我 'class' 是一个受保护的名称,不应用作变量。 Nano 不提供那些颜色,让我瞎了眼。

经验教训:孩子们,不要将 class 作为变量。