git_issue/insert_result_to_db.py

29 lines
938 B
Python

__author__ = 'qiangge'
import numpy as np
import pymysql
import csv
import helper
import dao
def get_analysis(proj_id,method):
path = 'result/'+proj_id+'/'+method+'/data/'
# get variance/different of each results
different = []
y_test = []
pred = []
x_id = []
for i in range(1,11,1):
probability = np.load(path + 'probability_' + repr(i) + ".npy")
y_test_temp = np.load(path + 'y_test_' + repr(i) + ".npy")
pred_temp = np.load(path + 'pred_' + repr(i) + ".npy")
x_id_temp = np.load(path + 'x_id_' + repr(i) + ".npy")
for j in range(len(probability)):
# diff = probability[j:j+1,0][0]-probability[j:j+1,1][0]
different.append(probability[j])
y_test.append(y_test_temp[j])
pred.append(pred_temp[j])
x_id.append(x_id_temp[j])
dao.insert_data(proj_id,x_id,different,pred,y_test)
get_analysis('524804','svm')