使用pickle将巨大的bigram词典保存到文件
问题内容:
我的一个朋友写了这个小程序。的textFile
是1.2GB大小(报纸7年价值)。他成功地创建了词典,但是无法使用pickle(程序挂起)将其写入文件。
import sys
import string
import cPickle as pickle
biGramDict = {}
textFile = open(str(sys.argv[1]), 'r')
biGramDictFile = open(str(sys.argv[2]), 'w')
for line in textFile:
if (line.find('<s>')!=-1):
old = None
for line2 in textFile:
if (line2.find('</s>')!=-1):
break
else:
line2=line2.strip()
if line2 not in string.punctuation:
if old != None:
if old not in biGramDict:
biGramDict[old] = {}
if line2 not in biGramDict[old]:
biGramDict[old][line2] = 0
biGramDict[old][line2]+=1
old=line2
textFile.close()
print "going to pickle..."
pickle.dump(biGramDict, biGramDictFile,2)
print "pickle done. now load it..."
biGramDictFile.close()
biGramDictFile = open(str(sys.argv[2]), 'r')
newBiGramDict = pickle.load(biGramDictFile)
提前致谢。
编辑
任何有兴趣的人,我将简要解释该程序的作用。假设您的文件格式大致如下:
<s>
Hello
,
World
!
</s>
<s>
Hello
,
munde
!
</s>
<s>
World
domination
.
</s>
<s>
Total
World
domination
!
</s>
<s>
是句子的分隔符。- 每行一个字。
会生成一个biGramDictionary,以供以后使用。
像这样的东西:
{
"Hello": {"World": 1, "munde": 1},
"World": {"domination": 2},
"Total": {"World": 1},
}
希望这可以帮助。现在,该策略已更改为使用mysql,因为sqlite无法正常工作(可能是因为大小)
问题答案: