-
Notifications
You must be signed in to change notification settings - Fork 1
/
genderclassmodelBYrandomforest.py
85 lines (54 loc) · 1.8 KB
/
genderclassmodelBYrandomforest.py
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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
import csv as csv
import numpy as np
from sklearn.ensemble import RandomForestClassifier
from collections import Counter
csv_file_objectg = csv.reader(open('genderclassmodel2.csv', 'rb')) #Load in the csv file
csv_file_objectf = csv.reader(open('myfirstforest2.csv', 'rb')) #Load in the csv file
csv_file_objectr = csv.reader(open('rbf.csv', 'rb')) #Load in the csv file
csv_file_objectn = csv.reader(open('normalizedforest.csv', 'rb')) #Load in the csv file
csv_file_objectgb = csv.reader(open('gbc.csv', 'rb')) #Load in the csv file
gdata = []
fdata = []
rdata = []
ndata = []
gbdata = []
avgdata = []
for row in csv_file_objectn:
ndata.append(row)
for row in csv_file_objectgb:
gbdata.append(row)
for row in csv_file_objectg:
gdata.append(row)
for row in csv_file_objectf:
fdata.append(row)
for row in csv_file_objectr:
rdata.append(row)
ndata = np.array(ndata)
gdata = np.array(gdata)
gbdata = np.array(gbdata)
fdata = np.array(fdata)
rdata = np.array(rdata)
avgsurvived = []
tempavg = ['1','2','3','4','5']
#tempavg = ['1','2','3']
print xrange(np.size(fdata[0::,0]))
print xrange(np.size(gdata[0::,0]))
for i in xrange(np.size(gdata[0::,0])):
tempavg[0] = gdata[i,0].astype(float)
tempavg[1] = fdata[i,0].astype(float)
tempavg[2] = rdata[i,0].astype(float)
tempavg[3] = gbdata[i,0].astype(float)
tempavg[4] = ndata[i,0].astype(float)
print tempavg
print np.round(np.mean(tempavg))
avgdata.append(np.round(np.mean(tempavg)))
open_file_object = csv.writer(open("avgforest.csv", "wb"))
test_file_object = csv.reader(open('test.csv', 'rb'))
test_file_object.next()
i = 0
for row in test_file_object:
row.insert(0,avgdata[i].astype(np.uint8))
open_file_object.writerow(row)
print row
print i
i += 1