SKlearn：通過讀取目錄中的多個文件來加載訓練數據

我可以從單個文件輸入測試數據，沒有任何問題。然而，每當我嘗試從目錄中的多個文件輸入數據時，我會得到以下錯誤：AttributeError：'NoneType'對象沒有屬性'lower'。請參閱下面的代碼，我會很感激任何幫助。謝謝。SKlearn：通過讀取目錄中的多個文件來加載訓練數據

from sklearn.feature_extraction.text import CountVectorizer 
from sklearn.feature_extraction.text import TfidfTransformer 
from nltk.corpus import stopwords 
import numpy as np 
import numpy.linalg as LA 

import os 
path = "C:\zircon" 

def radfil(): 
    for file in os.listdir(path): 
     current = os.path.join(path, file) 
     if os.path.isfile(current): 
      data = open(current, "rb").read() 
      print data 

train_set = [radfil()] 
test_set = ["The sun in the sky is bright."] 
stopWords = stopwords.words('english') 

vectorizer = CountVectorizer(stop_words=stopWords, min_df=1) 
#print vectorizer 
transformer = TfidfTransformer() 
#print transformer 

trainVectorizerArray = vectorizer.fit_transform(train_set).toarray() 
testVectorizerArray = vectorizer.transform(test_set).toarray() 
print 'Fit Vectorizer to train set', trainVectorizerArray 
print 'Transform Vectorizer to test set', testVectorizerArray

來源

2013-07-02 Tiger1

可以粘貼整個堆棧跟蹤嗎？ – Steve

我猜你的錯誤是由試圖在None類型的變量中執行lower（）操作引起的。也許這發生在

trainVectorizerArray = vectorizer.fit_transform(train_set).toarray()

radfil（）將返回一個None類型。嘗試組合來自文件的數據並向radfil（）添加一條return語句。這就是我沒有完整的堆棧跟蹤所能做的。

來源

2013-09-22 03:01:55 tweirick

SKlearn：通過讀取目錄中的多個文件來加載訓練數據

回答

相關問題