tensorflow - tensorflow中levenberg marquardt优化的实现

Question

我想检查 Levenberg Marquardt 优化是否已在 TensorFlow 中实现。我过去通常使用 Adam 优化器，但我读到 Levenberg Marquardt 优化对于非线性数据集效果更好。有人知道这在 Tensorflow 中是否可用吗？

Answer 1

是的，Levenberg marquardt 优化器在Tensorflow_Graphics 模块内的TensorFlow 中可用，并且可以使用tfg.math.optimizer 访问。 levenberg_marquardt。为此，您需要先安装 tensorflow-graphics。请检查以下代码:

!pip install tensorflow-graphics
import tensorflow_graphics as tfg
from tensorflow_graphics.math.optimizer import levenberg_marquardt