def buildTree(parser, depth, dir_name, filestart=None): if not filestart is None: tree = N.findFile(filestart, parser, dir_name=dir_name) else: tree = N.Node(directory=dir_name) old_index = parser.index while not parser.atEnd() and tree.utility.count < depth: inst = parser.getNext() N.addInc(tree, inst) if not filestart is None and parser.index >= old_index * 2: tree.saveObject(filestart+str(parser.index)+'.cbwb', remove=filestart+str(old_index)+'.cbwb') old_index = parser.index if not filestart is None: tree.saveObject(filestart+str(parser.index)+'.cbwb', remove=filestart+str(old_index)+'.cbwb') return tree
interpreter = argparse.ArgumentParser(
description="Show the surpisingness of phones already added to the cluster hierarchy."
)
interpreter.add_argument(
"saveName",
metavar="<begining_of_cbwb_file>",
type=str,
nargs=1,
help="The part of the .cbwb file for this cluster which comes before the '_' (i.e. \"test\" if the cluster is saved as test_N.cbwb)",
)
# interpreter.add_argument('-line', type=str, help='Show the line of average Surprise, Merges, Splits or Depth over time instead of regular output.')
interpreter.add_argument(
"-line",
action="store_true",
help="Show the line of average Surprise, Merges, Splits or Depth over time instead of regular output.",
)
interpreter.add_argument("-csv", type=str, help="Save surprise values to csv file.")
interpreter.add_argument(
"-unscale", action="store_true", help="Unscale the attribute values of centroids when generating dot pictures"
)
interpreter.set_defaults(showLine=False)
args = interpreter.parse_args()
tree = N.findFile(args.saveName[0] + "_", None)
if args.csv:
tree.viz.toCSV(args.csv)
if args.line:
tree.viz.surpriseLine()
else:
tree.viz.plotSurprise(unscale=args.unscale)
