用excel中的數據進行svr預測

import cv2
import numpy as np
import xlrd

def LoadPredictDataFromExcel():
    #文件位置
    ExcelFile=xlrd.open_workbook(r'..\0221_To_Teves訓練數據.xlsx')
#獲取目標EXCEL文件sheet名
    print (ExcelFile.sheet_names())

    sheet = ExcelFile.sheet_by_name('Sheet1')
#打印sheet的名稱,行數,列數
    print (sheet.name,sheet.nrows,sheet.ncols)
#獲取整行或者整列的值
    BackgroundDist = sheet.col_values(5)

    ForegroundDist = sheet.col_values(1)

    ForegroundArea = sheet.col_values(2)

    offset = sheet.col_values(3)

    BackgroundDist.pop(0)

    ForegroundDist.pop(0)

    ForegroundArea.pop(0)

    offset.pop(0)

    data = np.vstack((BackgroundDist, ForegroundDist, ForegroundArea))

    data = data.transpose()

    data = np.array(data,dtype='float32')

return data, offset

if __name__ == '__main__':
    data, offset  = LoadPredictDataFromExcel()

    SVR = cv2.ml.SVM_load('SVRModel.xml')#('NearModelPara.xml')

    result = SVR.predict(data)


    result = result[1].ravel()

    result = result.tolist()
print(result)
print(offset)
for i in range(len(result)):
        for j in range(len(offset)):
            flag = 0
            if offset[j] == result[i]:
                flag = 1
                break
        if flag == 0:
            print(result[i])

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章