def script(): filename = sys.argv[1] min_sup = sys.argv[2] D = graph_reader.graph_reader(filename) gspan.gSpan([], D, int(min_sup)) print "Total number of patterns: {}".format(gspan.num_call - 1)
minsup -- minimum support threshold (integer) """ output(C) E = right_most_path_extensions(C, D) for (t, sup) in E: C_prime = C + [t] sup_C_prime = sup if sup_C_prime >= minsup and is_canonical(C_prime): gSpan(C_prime, D, minsup) if __name__ == "__main__": # Test exampleG D = graph_reader.graph_reader("exampleG.txt") gSpan([], D, 2) # # Test Compound_422 # D = graph_reader.graph_reader("Compound_422.txt") # gSpan([], D, 100) # # Test right_most_path without support count # E = right_most_path_extensions(None, D) # for g in E.values(): # print g # # Test right_most_path with support count # E = right_most_path_extensions([], D) # for t, sup in E: # print "t: {}, sup: {}".format(t, sup)