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

2024-04-19 00:32:52 发布

您现在位置:Python中文网/ 问答频道 /正文

我一直在使用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分钟后会遇到内存错误。
任何人都可以向我解释如何增加python可以访问的内存,或者用其他方法来解决这个问题。在


Tags: pathtextinfromimportreforstring
2条回答

Python的内存限制不超过操作系统的要求。在

  • 请确保没有使用ulimit或等效工具限制进程的内存使用。在
  • 运行top并查看进程是否使用了所有可用内存。在
  • 然后你要么减少程序所需的内存,要么增加它访问的RAM/Swap。在

相关问题 更多 >