예제 #1
0
파일: final.py 프로젝트: SiuTo/DreamPC
def loadFinalData(flag):
	dataCSV = csv.reader(open("../data/data_"+flag+".csv", "r"))
	data = [row for row in dataCSV]
	formatData(data[1:], flag)
예제 #2
0
from format_data import formatData
from train_xgboost import train
import csv

dataCSV = csv.reader(open("../data/data_train.csv", "r"))
data = [row for row in dataCSV]
del data[0]
m = (len(data)-1)//10+1
data_origin = []
for i in range(len(data)):
	data_origin.append(data[i][:2])

for k in range(10):
	left, right = k*m, min((k+1)*m, len(data))
	realFile = open("real_value_"+str(k)+".csv", "w")
	lines = ["LKADT_P,DEATH\n"]
	for i in range(left, right):
		lines.append("{},{}\n".format(data[i][0], data[i][1]))
	realFile.writelines(lines)

	formatData(data[:left]+data[right:], "train")
	for i in range(left, right):
		data[i][0], data[i][1] = 0, 1
	formatData(data[left:right], "test")
	for i in range(left, right):
		data[i][0], data[i][1] = data_origin[i]

	train(str(k))