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