如何将字典转换为数组



我需要将分类模型的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)

当我运行这个代码非常感谢:)我得到这个错误:

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

X_train_pca是一个数组,所以我想我应该把lables_train也转换成数组,但我不知道怎么做,因为数组与列表不同。

试试这个

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

相关内容

  • 没有找到相关文章

最新更新