动画化神经网络的优化轨迹
loss-landscape-anim允许您在神经网络的损耗格局的2D切片中创建动画优化路径。 它基于 ,如果要添加自己的模型,请遵循其建议的样式。
请查看我的文章以获取更多示例和一些直观说明。
0.安装
从PyPI:
pip install loss-landscape-anim
从源头上讲,您需要。 克隆此存储库后,请运行以下命令以安装依赖项。
poetry install
1.基本范例
使用提供的和默认的多层感知器MLP模型,您可以直接调用loss_landscape_anim来获得示例动画GIF,如下所示:
# Use default MLP model and sample spirals dataset
loss_landscape_anim ( n_epochs = 300 )
注意:如果您在笔记本电脑上使用它,请不要忘记在顶部包括以
1