sklearn中测试及和训练集分离及选取

最新推荐文章于 2024-07-25 08:51:01 发布

ErrorMaker...

最新推荐文章于 2024-07-25 08:51:01 发布

阅读量334

点赞数

分类专栏：机器学习

本文链接：https://blog.csdn.net/qq_22667507/article/details/116062858

版权

机器学习专栏收录该内容

12 篇文章 0 订阅

订阅专栏

sklearn中的train_test_split：

from sklearn.model_selection import train_test_split

data_train,data_test,label_train,label_test = train_test_split(x,y,test_size=0.3,random_state=123)

test = KNeighborsClassifier(n_neighbors=4)
test.fit(data_train,label_train)
res1 = test.predict(data_test)
print(sum(res1 == label_test)/len(label_test))#准确率

#返回测试的准确率
from sklearn.metrics import accuracy_score
accuracy_score(res,test_label)