levmarq-torch

Levenberg-Marquardt算法的基本PyTorch实现。此算法解决形式为

$$\mathbf{x}^* = \mathrm{argmin}_{\mathbf{x}} |\mathbf{y} - \mathbf{\hat{y}}(\mathbf{x})|^2 , .$$

实现是对参数 $\mathbf{x}$ 和数据点 $\mathbf{y}$ 进行批处理的。

基于Gavin 2022的第1个实现和一些来自Connor Stone的帮助。

安装

运行

git clone git@github.com:adam-coogan/levmarq-torch.git
cd levmarq-torch
pip install .

将安装levmarq_torch包。