在PyMC3中恢复正弦函数参数

import pymc3 as pm
import numpy as np
import matplotlib.pyplot as plt

# 生成正弦函数的数据
np.random.seed(0)
x = np.linspace(0, 2*np.pi, 100)
y_true = np.sin(x)

# 添加噪声
y_obs = y_true + np.random.normal(0, 0.1, size=len(x))

with pm.Model() as model:
    # 定义参数的先验分布
    amplitude = pm.HalfNormal('amplitude', sd=1)
    phase = pm.Uniform('phase', lower=0, upper=2*np.pi)
    offset = pm.Normal('offset', mu=0, sd=1)

    # 定义正弦函数模型
    y_pred = amplitude * pm.math.sin(x + phase) + offset

    # 定义观测数据的似然函数
    likelihood = pm.Normal('likelihood', mu=y_pred, sd=0.1, observed=y_obs)

with model:
    # 使用MCMC方法进行推断
    trace = pm.sample(2000, tune=1000)

# 绘制参数的后验分布
pm.plot_posterior(trace, var_names=['amplitude', 'phase', 'offset'])
plt.show()