mlflow.system_metrics

系统指标记录模块。

mlflow.system_metrics.disable_system_metrics_logging()[source]

全局禁用系统指标日志记录。

调用此函数将全局禁用系统指标日志记录,但用户仍可通过 mlflow.start_run(log_system_metrics=True) 为单个运行选择加入系统指标日志记录。

mlflow.system_metrics.enable_system_metrics_logging()[source]

全局启用系统指标日志记录。

调用此函数将全局启用系统指标日志记录,但用户仍可通过 mlflow.start_run(log_system_metrics=False) 为单个运行选择退出系统指标日志记录。

mlflow.system_metrics.set_system_metrics_node_id(node_id)[source]

设置系统指标节点 ID。

node_id 是收集指标的机器的标识符。这在多节点(分布式训练)设置中很有用。

mlflow.system_metrics.set_system_metrics_samples_before_logging(samples)[source]

设置记录系统指标之前的采样数。

每次收集了 samples 个样本后,系统指标将记录到 mlflow。默认情况下 samples=1

mlflow.system_metrics.set_system_metrics_sampling_interval(interval)[source]

设置系统指标采样间隔。

interval 秒,将收集系统指标。默认情况下 interval=10