|
@@ -274,12 +274,12 @@ if __name__ == "__main__":
|
|
|
print("vtc vocs: {}".format(np.sum(data['vtc'])))
|
|
|
print("duration: {}".format(duration))
|
|
|
|
|
|
- with open(f'data_{args.output}.pickle', 'wb') as fp:
|
|
|
+ with open(f'output/samples/data_{args.output}.pickle', 'wb') as fp:
|
|
|
pickle.dump(data, fp, pickle.HIGHEST_PROTOCOL)
|
|
|
|
|
|
posterior = stan.build(stan_code, data = data)
|
|
|
fit = posterior.sample(num_chains = args.chains, num_samples = args.samples)
|
|
|
df = fit.to_frame()
|
|
|
- df.to_parquet(f'fit_{args.output}.parquet')
|
|
|
+ df.to_parquet(f'output/samples/fit_{args.output}.parquet')
|
|
|
|
|
|
|