def extract_hier_all(gosubdag, out, root_term, go2geneids, prize_dictionary): """write_hier.py: Prints the entire mini GO hierarchy, with counts of children.""" out.write('\nTEST EXTRACTION: Print all hierarchies:\n') objwr = WrHierGO(gosubdag, go2geneids=go2geneids, prize_dictionary=prize_dictionary, G=nx.DiGraph()) obj = objwr.ext_hier_down(root_term, out) seeds = prize_dictionary.keys() layer_i_nodes = linear_threshold(obj.G, seeds, steps=INFLUENCE_N_STEPS) return (obj.vertices, obj.edges, obj.G, layer_i_nodes)
def extract_hier_all(gosubdag, out, root_term, go2geneids): """write_hier.py: Prints the entire mini GO hierarchy, with counts of children.""" out.write('\nTEST EXTRACTION: Print all hierarchies:\n') objwr = WrHierGO(gosubdag, go2geneids=go2geneids) obj = objwr.ext_hier_down(root_term, out) return (obj.vertices, obj.edges)