2016-06-28 57 views
1

我試圖在scikit-learn 網站http://scikit-learn.org/stable/tutorial/text_analytics/working_with_text_data.html中探索此示例的不同分類器。但是,下面的代碼產生了一個錯誤:ValueError:用一個序列設置一個數組元素。python錯誤設置序列的數組元素

from sklearn.feature_extraction.text import CountVectorizer 
from sklearn.feature_extraction.text import TfidfTransformer 
import tensorflow.contrib.learn as skflow 

data = ["I so handsome. I just broke the mirror!","I am a normal guy."] 
label = np.array([0,1]) 

#CountVectoriser 
count_vect = CountVectorizer() 
X_train_counts = count_vect.fit_transform(data) 

#TfidfTransformer 
tfidf_transformer = TfidfTransformer() 
X_train_tfidf = tfidf_transformer.fit_transform(X_train_counts) 

#Classifier 
clf = skflow.TensorFlowLinearClassifier(n_classes=2) 
clf.fit(X_train_tfidf, label) 

回答

2

TensorFlowLinearClassifier不處理企業社會責任矩陣作爲輸入,你可以按照that issue進展。


你可以現在做的是將其送入clf.fit()之前轉換X_train_tfidf爲numpy的矩陣:

clf.fit(X_train_tfidf.toarray(), label) 
相關問題