parser.add_argument('--nsamples', metavar='ns', help='The number of samples')
args = parser.parse_args()
""" 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)
