1、决策树分类模型
from sklearn.datasets import load_iris
from sklearn import tree
from sklearn.model_selection import train_test_split
tree = tree.DecisionTreeClassifier(criterion='entropy')
X = load_iris().data
Y = load_iris().target
x_train, x_test, y_train, y_test = train_test_split(X,Y, test_size = 0.3, random_state = 0)
tree.fit(x_train,y_train)
print("模型训练集的准确率:%.3f" %tree.score(x_train, y_train))
print("模型测试集的准确率:%.3f" %tree.score(x_test, y_test))
2、KNN分类模型
from sklearn.datasets import load_iris
from sklearn import neighbors
from sklearn.model_selection import train_test_split
KNN = neighbors.KNeighborsClassifier(n_neighbors = 5)
X = load_iris().data
Y = load_iris().target
x_train, x_test, y_train, y_test = train_test_split(X,Y, test_size = 0.3,
random_state = 0)
KNN.fit(x_train,y_train)
print("KNN模型训练集的准确率:%.3f" %KNN.score(x_train, y_train))
print("KNN模型测试集的准确率:%.3f" %KNN.score(x_test, y_test))
3、朴素贝叶斯分类模型
from sklearn.datasets import load_iris
from sklearn import naive_bayes
from sklearn.model_selection import train_test_split
bayes = naive_bayes.GaussianNB()
X = load_iris().data
Y = load_iris().target
x_train, x_test, y_train, y_test = train_test_split(X,Y, test_size = 0.3,
random_state = 0)
bayes.fit(x_train,y_train)
print("贝叶斯模型训练集的准确率:%.3f" %bayes.score(x_train, y_train))
print("贝叶斯模型测试集的准确率:%.3f" % bayes.score(x_test, y_test))
4、SVM分类模式
from sklearn.datasets import load_iris
from sklearn import neighbors
from sklearn import svm
from sklearn.model_selection import train_test_split
svm = svm.SVC(gamma ='auto')
X = load_iris().data
Y = load_iris().target
x_train, x_test, y_train, y_test = train_test_split(X,Y, test_size = 0.3,
random_state = 0)
svm.fit(x_train,y_train)
print("SVM模型训练集的准确率:%.3f" %svm.score(x_train, y_train))
print("SVM模型测试集的准确率:%.3f" %svm.score(x_test, y_test))