""" Initial Workflow definition """ if args.path: sc = SparkContext() data = nbl.load(args.path).get_data()[:,:,:,:100] flow = Workflow(data, sc).extract() op = args.operator if op == "vb": flow = flow.visualizeBrain() elif op == "v": nsamples = args.nsamples flow = flow.visualize(nsamples=int(nsamples)) elif op == "vc": k = args.k flow = flow.clustering(int(k)).visualize() elif op == "ts": k = args.k flow = flow.clustering(int(k)) flow.execute() with open("model", "a+") as output: pickle.dump(flow.last.result, output, pickle.HIGHEST_PROTOCOL) exit("Model Saved") elif op == "pr": utils.predict(args.model, args.vector)