使用sklearn特征提取时出现内存错误

时间:2014-11-23 14:27:10

标签: python python-3.x scikit-learn nltk

我一直在使用python脚本来标记和计算很多.txt文件的TFIDF,我的脚本如下:

import nltk
import string
import os

from sklearn.feature_extraction.text import TfidfVectorizer
from nltk.stem.porter import PorterStemmer
from nltk.corpus import stopwords
import nltk
import string
from collections import Counter
from sklearn.feature_extraction.text import CountVectorizer
from sklearn.externals import joblib
import re
import scipy.io
import glob

path = 'R'
token_dict = {}
stemmer = PorterStemmer()

def stem_tokens(tokens, stemmer):
    stemmed = []
    for item in tokens:
        stemmed.append(stemmer.stem(item))
    return stemmed

def tokenize(text):
    tokens = nltk.word_tokenize(text)
    stems = stem_tokens(tokens, stemmer)
    return stems

for subdir, dirs, files in os.walk(path):
 for file in files:
    #if re.match("text\d+.txt",file):
      #with open(os.path.join(path,file),'r') as f:
       #for shakes in f:

        remove_spl_char_regex = re.compile('[%s]' % re.escape(string.punctuation)) # regex to remove special characters
        remove_num = re.compile('[\d]+')
        file_path = subdir + os.path.sep + file
        shakes = open(file_path, encoding="utf8")

        text = shakes.read()
        lowers = text.lower()
        a1 = lowers.translate(string.punctuation)
        a2 = remove_spl_char_regex.sub(" ",a1)  # Remove special characters
        a3 = remove_num.sub("", a2)  #Remove numbers
        token_dict[file] = a3

tfidf = TfidfVectorizer(tokenizer=tokenize, stop_words='english')
tfs = tfidf.fit_transform(token_dict.values())
scipy.io.savemat('arrdata4.mat', mdict={'arr': tfs})

根据文件的大小,我在30分钟后遇到MemoryError 任何人都可以向我解释如何增加python可以访问的内存或我可以解决此问题的任何其他方式? 。

1 个答案:

答案 0 :(得分:1)

Python没有超出操作系统强加的内存限制。

  • 请确保您不会使用ulimit或同等版本限制进程的内存使用量。
  • 运行top并查看该进程是否使用了所有可用内存。
  • 然后你要么必须减少你的程序需要的内存,要么增加它可以访问的RAM / Swap。