def loadFinalData(flag):
dataCSV = csv.reader(open("../data/data_"+flag+".csv", "r"))
data = [row for row in dataCSV]
formatData(data[1:], flag)
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))