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
因此该函数会写入&#34; Id,Sales&#34;在submission.txt中,但从未完成for循环... 谢谢您的帮助。我真的不知道该怎么做。
答案 0 :(得分:1)
你的循环可能只花了很长时间来迭代遍历SFrame。错误消息是因为您使用Ctrl-C取消了命令(我在猜测)。
如果你想用这两列写一个csv,试试这个(假设var是一个SFrame):
var[['Id','Prediction']].save("submission", format="csv")