【cmdstanpy】Stanのフィッティング後のモデルを保存する【chatGPTメモ】

cmdstanpyで事後分布を計算した後保存したいなと思ったら、そのままcsvに行けるらしい。今度試してみる。

 

import cmdstanpy
import pandas as pd

# Stanモデルのコンパイル
model = cmdstanpy.CmdStanModel(stan_file='model.stan')

# データの準備
data = {'N': 100, 'x': [1, 2, 3, ..., 98, 99, 100], 'y': [2, 4, 6, ..., 196, 198, 200]}

# サンプリングの実行
fit = model.sample(data=data)

# データの保存
fit.save_csvfiles(dir='fit_output')

# データの読み込み
csv_files = cmdstanpy.from_csv('fit_output')
summary = csv_files.summary()
print(summary)