使用 Scikit-learn 的支持向量机分类鸢尾花数据集

from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
from sklearn.svm import SVC

# 加载数据集
iris = load_iris()
X = iris.data[:, :2]
y = iris.target

# 分割数据集
c = float(input('请输入训练集比例 (例如: 0.6): '))
X_train, X_test, y_train, y_test = train_test_split(X, y, train_size=c, random_state=1)

# 训练模型
model = SVC(kernel='rbf', gamma=20, decision_function_shape='ovo')
model.fit(X_train, y_train)

# 计算训练集和测试集的精度
train_accu = model.score(X_train, y_train)
test_accu = model.score(X_test, y_test)

# 输出精度结果
print('accu of model for train set is {:.2f}'.format(train_accu))
print('accu of model for test set is {:.2f}'.format(test_accu))