机器学习--sklearn的常见使用

最新推荐文章于 2024-09-14 14:55:50 发布

原创最新推荐文章于 2024-09-14 14:55:50 发布 · 1.1k 阅读

0 ·

CC 4.0 BY-SA版权

机器学习专栏收录该内容

13 篇文章

订阅专栏

本文介绍了如何使用sklearn库进行机器学习，特别是朴素贝叶斯和SVM的支持向量机。首先，通过GaussianNB实现朴素贝叶斯模型，训练并预测数据，计算预测准确率。接着，使用SVC构建支持向量机模型，调整C参数和核函数，展示了训练和预测的时间消耗，并得出预测准确率。

# 朴素贝叶斯
from sklearn.naive_bayes import GaussianNB
clf = GaussianNB()          #训练模型
clf.fit(features_train,labels_train)     
pred = clf.predit(features_test)        #给出测试变量，预测结果

from sklearn.metrics import accuracy_score
accuracy = accuracy_score(pred, labels_test)   # 预测准确率

#-------------------------------------------------------------------------------------------
# SVM支持向量机，可以通过减少训练集的大小来减少运行时间（牺牲准确率）
from sklearn.svm import SVC

clf = SVC(C=10000,kernel='rbf')     # C的大小决定拟合程度，kernel是核函数，有linear\rbf\polynomial等
t0 = time()                          # C越大越好，但是可能会过拟合
clf.fit(features_train,labels_train)
print "train time is :" , round(time()-t0, 3), "s"    

t0 = time()
pred = clf.predict(features_test)
print "fit time is :" , round(time()-t0,3), "s"
#print pred[10],pred[26],pred[50]               # 得出第10/26/50的预测结果

from sklearn.metrics import accuracy_score
acc = accuracy_score(pred, labels_test)
print acc                                      # 准确率

#--------------------------------------------------------------------------------------------