MLflow

class sklearn_genetic.mlflow_log.MLflowConfig(tracking_uri, experiment, run_name, save_models=False, registry_uri=None, tags=None)[源代码]

在mlflow运行实例中记录每次超参数拟合:https://mlflow.org/

Parameters:
tracking_uri: str

本地或远程跟踪服务器的地址。

experiment: str

要激活的实验名称(区分大小写)。

run_name: str

新运行的名称(存储为mlflow.runName标签)。

save_models: bool, default=False

如果 True,会将评估器记录到mlflow工件中

registry_uri: str, default=None

本地或远程模型注册服务器的地址。

tags: dict, default=None

标签名字典:字符串 -> 值。

create_run(parameters, score, estimator)[源代码]
Parameters:
parameters: dict

一个字典,其中键为超参数名称,值为当前参数设置值

score:

当前参数达到的交叉验证分数

estimator: estimator object

当前正在拟合的sklearn估计器