3

Python 未完成以下功能（使用 iPython Notebook 和 Graphlab）

def print_submission_file(var, filename='submission.txt'):
    with open(filename, 'w') as f:
        f.write('Id,Sales\n')
        for row in var:
            f.write(str(row['Id']) + ',' + str(row['Prediction']) + '\n')

print_submission_file(test, 'submission.txt')

这是脚本停止时的日志（5/10 分钟后）

---------------------------------------------------------------------------
RuntimeError                              Traceback (most recent call last)
<ipython-input-9-f51bf8359fb9> in <module>()
  5             f.write(str(row['Id']) + ',' + str(row['Prediction']) + '\n')
  6 
----> 7 print_submission_file(test, 'submission.txt')

<ipython-input-9-f51bf8359fb9> in print_submission_file(var, filename)
  2     with open(filename, 'w') as f:
  3         f.write('Id,Sales\n')
----> 4         for row in var:
  5             f.write(str(row['Id']) + ',' + str(row['Prediction']) + '\n')
  6 

/Users/MaximeDeBruyn/.graphlab/anaconda/lib/python2.7/site-packages/graphlab/data_structures/sframe.pyc in generator()
   3694         def generator():
   3695             elems_at_a_time = 262144
-> 3696             self.__proxy__.begin_iterator()
   3697             ret = self.__proxy__.iterator_get_next(elems_at_a_time)
   3698             column_names = self.column_names()

graphlab/cython/cy_sframe.pyx in        graphlab.cython.cy_sframe.UnitySFrameProxy.begin_iterator()

graphlab/cython/cy_sframe.pyx in graphlab.cython.cy_sframe.UnitySFrameProxy.begin_iterator()

RuntimeError: Runtime Exception. Canceled by user

因此，该函数在 submit.txt 中写入“Id，Sales”，但从未完成 for 循环......谢谢您的帮助。我真的不知道该怎么办。

4

1 回答 1

1

您的循环可能只需要很长时间才能以这种方式遍历 SFrame。错误消息是因为您使用 Ctrl-C 取消了命令（我猜）。

如果你想用这两列写一个 csv，试试这个（假设 var 是一个 SFrame）：

var[['Id','Prediction']].save("submission", format="csv")
于 2015-10-11T20:50:04.770 回答