机器学习基础 互动版

在线工具推荐: Three.js AI纹理开发包 - YOLO合成数据生成器 - GLTF/GLB在线编辑 - 3D模型格式在线转换 - 可编程3D场景编辑器

支持向量机聚类训练

将训练集数据输入支持向量机进行训练,其中鸢尾花的各项属性为输入值,鸢尾花的种类编号为输出值进行训练。

from sklearn import datasets            #鸢尾花数据集被sklearn的datasets所包含,需要引用
from sklearn.cross_validation import train_test_split
from sklearn.svm import SVC
from numpy import *

iris_dataset = datasets.load_iris();    #加载鸢尾花数据集    
iris_data = iris_dataset.data;            
iris_target = iris_dataset.target;
x_train,x_test,y_train,y_test = train_test_split(iris_data,iris_target,test_size=0.2);
SVC_0 = SVC(kernel='rbf');
'''
创建支持向量机分类器,其中核函数为rbf(径向基函数)
除此之外,还有linear(线性核函数)、poly(多项式核函数模型)
'''
SVC_0.fit(x_train,y_train);
'''
将训练集数据输入模型进行训练
'''