Negative_Search_Algo_Genetic_Algo/Create_data_file.py at master · silversword05/Negative_Search_Algo_Genetic_Algo · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46


f_dat = open("Fisher.txt","r")
lns_dat = f_dat.readlines()

max_label = 3
no_of_train = 100
no_of_test = 50

f_in = open("data_input_train.txt","w")
f_out = open("data_output_train.txt","w")

for lns in lns_dat[:no_of_train]:
    intgs = [int(x) for x in lns.strip(' \n').split('\t')]
    for j in range(max_label):
        if(intgs[0] == j):
            f_out.write("1 ")
        else:
            f_out.write("0 ")
    f_out.write("\n")

    intgs = intgs[1:]
    f_in.write(" ".join(str(x) for x in intgs))
    f_in.write("\n")

f_in.close()
f_out.close()

f_in = open("data_input_test.txt","w")
f_out = open("data_output_test.txt","w")

for lns in lns_dat[no_of_train:]:
    intgs = [int(x) for x in lns.strip(' \n').split('\t')]
    for j in range(max_label):
        if(intgs[0] == j):
            f_out.write("1 ")
        else:
            f_out.write("0 ")
    f_out.write("\n")

    intgs = intgs[1:]
    f_in.write(" ".join(str(x) for x in intgs))
    f_in.write("\n")

f_in.close()
f_out.close()