2016-03-29 10 views
-3

私の分類モデルのlabels_train(辞書のリストは[{Id:0, Prediction:2},{Id:1, Prediction:4},....])を配列に変換する必要があります。辞書を配列に変換するには

#Classification model: 
from sklearn.linear_model import LogisticRegression 
class_model = LogisticRegression() 

#Fitting the classifier to the training set. 
classifier= class_model.fit(X_train_pca, labels_train) 
score= classifier.score(X_test_pca, labels_test) 
print(score) 

私はこのcodeThankを実行すると、あなたは非常に:)私はこのエラーを取得する:

TypeError: unorderable types: dict()>dict() 

X_train_pcaが配列であるので、「私は、私は配列にもlables_trainを変換する必要があることを考えたが、私はドン配列がリストと異なるため、それを行う方法はわかりません。

+1

をお試しください)。 – timgeb

答えて

1

私たちにあなたがしようとしたと定義したものを「アレイ」(すなわちnumpyの配列やPythonのリストを教えて、あなたの実際のコードから問題とその性質を切り離す、入力、期待される出力を提供します?この

class_model.fit(X_train_pca, [i['Prediction'] for i in labels_train]) 
+0

これは動作します!私はそれを試しましたが、別の文脈で:-D –

+0

ありがとう! – Mary