test_with_3_variables.py
525 Bytes
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
import pandas as pd
#load the data
data = pd.read_csv("test_three_variables.csv")
names = data.columns.values.tolist()
n, m = data.shape
import myGraph
import score
#load the arc data
arc = pd.read_csv("arc.csv")
score_bde = []
for i in xrange(25):
graph = myGraph.Graph()
graph.add_vertice(range(m),label = names)
tmp = arc.ix[arc.graph == (i + 1)]
f = tmp['from'].tolist()
t = tmp['to'].tolist()
graph.add_edge(f,t)
myscore = score.score(graph, data)
score_bde.append(myscore.BDe())
print i+1