# -*- coding:UTF-8 -*-
from sklearn import datasets
from sklearn import svm
from sklearn.model_selection import train_test_split
from numpy import loadtxt
import pandas as pd
import numpy as np
import pickle
'''
#读取鸢尾花数据集
iris = datasets.load_iris()#读取鸢尾花数据集
# 读取特征
iris_feature = iris.data
# 读取标签
iris_label = iris.target
# 划分训练集与测试集
X_train, X_test, Y_train, Y_test = train_test_split(iris_feature, iris_label, test_size=0.3, random_state=42)
svm_classifier = svm.SVC(C=1.0, kernel='linear', decision_function_shape='ovr', gamma=0.01)
#svm_classifier = svm.LinearSVC(penalty='l2', loss='squared_hinge', dual=True, tol=0.0001, C=1.0, multi_class='ovr', fit_intercept=True, intercept_scaling=1, class_weight=None, verbose=0, random_state=None, max_iter=1000)
svm_classifier.fit(X_train, Y_train)
print("train", svm_classifier.score(X_train, Y_train))
print("test:", svm_classifier.score(X_test, Y_test))
print svm_classifier.coef_
print svm_classifier.intercept_
'''
'''
df = pd.read_csv('xxx.csv')
#x为去掉第一列的所有内容,第一列为label
x = df.iloc[:, 1:]
x=np.array(x)
y =df.loc[:,'type']
y=np.array(y)
X_train=x
Y_train=y
#X_train, X_test, Y_train, Y_test = train_test_split(x, y, test_size=0.5, random_state=42)
svm_classifier = svm.SVC(C=1.0, kernel='linear', decision_function_shape='ovr', gamma=0.01)
svm_classifier.fit(X_train, Y_train)
#print("train", svm_classifier.score(X_train, Y_train))
#print("test:", svm_classifier.score(X_test, Y_test))
print svm_classifier.coef_
print svm_classifier.intercept_
#print svm_classifier.support_vectors_
'''
#保存训练好的模型
#s=pickle.dumps(svm_classifier)
#f=open('svm.txt','w')
#f.write(s)
#f.close()
#读取训练好的模型
f2=open('svm.txt','r')
s2=f2.read()
clf=pickle.loads(s2)
print clf.coef_
print clf.intercept_
print clf.support_vectors_
#l=[0,1,0,3.661932872,0.384615385,26]
#l=np.array(l)
#print clf2.predict([l])
#tx=[1,1,1,1.056641667,0,2]
#tx=np.array(tx)
#print svm_classifier.predict([tx])
初识SVM
最新推荐文章于 2022-08-31 08:58:05 发布