1、导入所需库
from sklearn import svm
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
from sklearn.metrics import confusion_matrix
from sklearn.metrics import accuracy_score
import matplotlib.pyplot as plt
2、加载数据集(以iris为例)
# load iris dataset
iris = load_iris()
n_samples, n_features = iris.data.shape
print((n_samples, n_features))
print(iris.keys())
print(iris.target)
# split train test
train_data, test_date = train_test_split(iris.data, random_state=1, train_size=0.7, test_size=0.3)
train_label, test_label = train_test_split(iris.target, random_state=1, train_size=0.7, test_size=0.3)