机器学习实验管理 (ML Experiment Management)#

https://www.jarvis73.com/2020/11/19/ML-Experiment-Management/

我们下面讨论九个机器学习实验管理工具. (点击Logo进入项目主页)

编号

框架

年份

开源

公司

特性

1

TensorBoard

2017

Github

Google Inc.

跟踪实验指标, 可视化模型图, 嵌入, 语音展示, 图像展示, 其他插件

2

Sacred

2014

Github

IDSIA

配置、组织、记录和复制计算实验, 开销小, 鼓励实验的模块化和可配置性

3

Neptune

2016

Github

Neptune Labs Inc.

轻量级的实验管理工具

4

LossWise

2018

-

Mathpix Inc.

简单的API, 记录参数、图像、日志, 整合统计指标

5

Guild

2019

Github

TensorHub Inc.

跟踪实验, 自动管线, 微调参数

6

Allegro

2019

Github

Allegro AI

端到端企业级平台, 可管理机器学习和深度学习产品的生命周期

7

mlflow

2018

Github

LF Proj. LLC.

兼容任意机器学习库、语言, 可多人协作, 可以拓展到Spark

8

Weights & Biases

2017

-

Weights and Biases Inc.

跟踪、比较、可视化机器学习实验, 可协作, 企业支持

9

Comet

2017

-

Comet ML Inc.

跟踪、比较、解释和优化实验和模型

功能

TensorBoard

Sacred

Neptune

LossWise

Guild

Allegro

mlflow

W & B

Comet

标量

比较

资源信息

资源监控

源码

计算图

二进制

图像

数据分布

嵌入

音频

后端

MongoDB
Neptune*

前端

Omniboard
Neptune*

✔TensorBoard

前端评级

✩✩

✩✩

✩✩✩

✩✩✩

✩✩✩✩

✩✩

✩✩✩✩✩

✩✩✩✩✩

参数管理

✩✩✩✩✩

✩✩✩

✩✩

✩✩

✩✩

✩✩✩

✩✩✩

云服务

云空间

100G

?

100G

100G

本地部署

✔$

团队协作

插件

参数调优

商业支持

✔$

✔$

内部支持的框架

Tensorflow
PyTorch

Keras
Tensorflow
PyTorch
Sacred
mlflow*

Keras

Keras
Tensorflow
PyTorch
sklearn
MXNet
XGBoost

Keras
Tensorflow
PyTorch
sklearn
MXNet
Caffe2*

Keras
Tensorflow
PyTorch
sklearn
XGBoost*

Keras
Tensorflow
PyTorch
sklearn
XGBoost*

Keras
Tensorflow
PyTorch
sklearn
MXNet
Caffe2*