预测一种汽车并将结果连接到 ID

Question

我正在建立一个预测模型来判断一辆汽车是否是跑车。该模型工作正常，但我想将预测值加入到唯一 ID 并可视化比例等。基本上我有两个数据帧：

使用标记数据进行测试 - test_cars

CarId	Feature1	Feature2	IsSportCar
1	90	150	True
2	60	200	False
3	560	500	True

要预测的未标记数据 - cars_new

CarId	Feature1	Feature2
4	88	666
5	55	458
6	150	125

from sklearn.neighbors import KNeighborsClassifier

# Create arrays for the features and the response variable
y = test_cars['IsSportCar'].values
X = test_cars.drop(['IsSportCar','CarId'], axis=1).values

X_new = cars_new.drop(['CarId'], axis=1).values

# Create a k-NN classifier with 10 neighbors
knn = KNeighborsClassifier(n_neighbors=10)

# Fit the classifier to the data
knn.fit(X,y)

y_pred = knn.predict(X_new)

该模型运行良好，但我想将预测值加入到每辆汽车 (CarId) 中，因此 car_new 数据帧将与预测列“IsSportCar”一起输出：

CarId	Feature1	Feature2	IsSportCar
4	88	666	False
5	55	458	True
6	150	125	True

关于如何将预测值连接回唯一 ID 的任何想法？

Answer 1

cars_new['IsSportCar'] = y_pred

我假设 y_pred 是您要放入 cars_new 的变量？

预测一种汽车并将结果连接到 ID

predict a type of car and connect the results to IDs

python

pandas

scikit-learn