k_type = Type.TERMINAL if k_id.startswith("AKT") or k_id.startswith("G6P") else Type.NORMAL
s_id = str(row["K.ID"])
s_type = Type.TERMINAL if s_id.startswith("AKT") or s_id.startswith("G6P") else Type.NORMAL
network.create_node(k_id, 3, k_type)
network.create_node(s_id, 3, s_type)
print "Nodes created! Elapsed time: ", str(int(time.time() - start)), "s"
# Create edges
for index, row in ks_network.iterrows():
k_id = str(row["S.ID"])
s_id = str(row["K.ID"])
network.create_edge(k_id, s_id, 2)
print "Edges created! Elapsed time: ", str(int(time.time() - start)), "s"
print network
# Convert to directed graph
network.convert_to_directed(10)
print "Bi-graph created! Elapsed time: ", str(int(time.time() - start)), "s"
# Formulate PCST
pcst = PCST(network, 1)
pcst.initialise()
print "LP created! Elapsed time: ", str(int(time.time() - start)), "s"
network.create_node('6', 2)
network.create_node('7', 2)
network.create_node('8', 2)
network.create_node('9', 2)
network.create_node('10', 2)
network.create_node('11', 2)
network.create_node('12', 2)
# Target nodes
network.create_node('a', 2, Type.TERMINAL)
network.create_node('b', 2, Type.TERMINAL)
network.create_node('c', 2, Type.TERMINAL)
network.create_node('d', 2, Type.TERMINAL)
# Edges
network.create_edge('C', '5', 2)
network.create_edge('5', '1', 2)
network.create_edge('1', '2', 3)
network.create_edge('2', '6', 5)
network.create_edge('6', 'A', 3)
network.create_edge('1', '4', 2)
network.create_edge('4', 'a', 2)
network.create_edge('4', 'b', 2)
network.create_edge('1', '3', 4)
network.create_edge('3', 'B', 2)
network.create_edge('B', 'b', 4)
network.create_edge('9', 'B', 4)
network.create_edge('3', '9', 4)
network.create_edge('1', 'A', 1)
network.create_edge('A', '7', 5)
network.create_edge('9', 'A', 1)
