如何用Pymc3实现因子分解机？

Question

the output of FM. 如何使用pymc3实现FM模型？ 通常，我很难自定义输出Y的分布

我曾经从：https://github.com/dockerizeme/dockerizeme/blob/d651393988c5fc87f15553d20f4930a97fcfbcfd/hard-gists/00a17b1d374dfc267a9a/snippet.py学习如何使用pymc3实现bpmf

Y=pm.Noraml（'Y'，mu=w_0+np.dot（w，X）+0，tau=alpha*np.ones（（n，1）），观察值=标签）