上传者: 42169245
|
上传时间: 2022-03-08 16:47:18
|
文件大小: 73.14MB
|
文件类型: -
动画化神经网络的优化轨迹
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 )
注意:如果您在笔记本电脑上使用它,请不要忘记在顶部包括以