【原】用excel中的数据进行svr训练

a412841321

于 2019-03-01 11:42:25 发布

阅读量1k

点赞数

分类专栏： ml

本文链接：https://blog.youkuaiyun.com/a412841321/article/details/90609165

版权

import cv2
import numpy as np
import xlrd
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
from matplotlib.colors import ListedColormap

def svm_config():
    svm = cv2.ml.SVM_create()

    svm.setCoef0(0)

    svm.setCoef0(0.0)

    svm.setDegree(3)

    criteria = (cv2.TERM_CRITERIA_MAX_ITER + cv2.TERM_CRITERIA_EPS, 1000, 1e-3)#(cv2.TERM_CRITERIA_EPS, 100, 5e-3)
    svm.setTermCriteria(criteria)

    svm.setGamma(0.01)#0)
    svm.setKernel(cv2.ml.SVM_RBF)#)SVM_LINEAR
    svm.setNu(0.5)

    svm.setP(5e-3)#0.1)
    svm.setC(1)#0.01)
    svm.setType(cv2.ml.SVM_EPS_SVR)

return svm

#svm训练
def svm_train(svm,features,labels):
    # svm.train(np.array(features,dtype='float32'),cv2.ml.ROW_SAMPLE,np.array(labels,dtype='float32'))
    svm.trainAuto(np.array(features,dtype='float32'),cv2.ml.ROW_SAMPLE,np.array(labels,dtype='float32&#