-
Notifications
You must be signed in to change notification settings - Fork 2
/
data_loader.py
48 lines (39 loc) · 1.26 KB
/
data_loader.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
import csv
class DataLoader:
def write_to_disk(self, writer,typ,x,y):
writer.writerow(["x","y","type"])
for i in range(len(x)):
writer.writerow([x[i],y[i],typ])
def read_csv(self,filename):
x = []
y = []
spamReader = csv.reader(open("data/"+filename,"rb"))
spamReader.next()
for row in spamReader:
x.append(float(row[0]))
y.append(float(row[1]))
return x,y
def get_data(self):
y_all = []
x, y = self.read_csv("apple_out_rel.csv")
x, y1 = self.read_csv("bmw_out_rel.csv")
x, y2 = self.read_csv("google_out_rel.csv")
x, y3 = self.read_csv("goldman_out_rel.csv")
x, y4 = self.read_csv("shell_out_rel.csv")
x, y5 = self.read_csv("agu_out_rel.csv")
x, y6 = self.read_csv("staples_out_rel.csv")
x, y7 = self.read_csv("yahoo_out_rel.csv")
x, y8 = self.read_csv("cabot_out_rel.csv")
x, y9 = self.read_csv("france_telekom_out_rel.csv")
y_all.append(y5)
y_all.append(y)
y_all.append(y8)
y_all.append(y1)
y_all.append(y9)
y_all.append(y2)
y_all.append(y3)
y_all.append(y4)
y_all.append(y6)
y_all.append(y7)
labels = ["Agu", "Apple", "Cabot", "BMW", 'France Telekom', 'Google', 'Goldman', 'Shell', "Staples", "Yahoo"]
return x,y_all, labels