AutoMLAbstractRunContext 类

AutoML 运行上下文的包装器类。

继承

ABC

AutoMLAbstractRunContext

构造函数

AutoMLAbstractRunContext()

方法

batch_save_artifacts	将一批文本文件和模型保存到 Artifacts 中。 这比保存单个项目更高效。
get_run	生成运行上下文。 通过 contextlib 进行包装，以将其转换为上下文管理器。 嵌套调用将返回相同的运行上下文。
save_model_output	使用此运行上下文将给定的拟合模型保存到给定路径。
save_onnx_model_output	使用此运行上下文将给定的 onnx 模型保存到给定的远程路径。
save_onnx_model_resource	使用此运行上下文将给定的 onnx 模型资源保存到给定的远程路径。
save_str_output	将 str 文件作为 txt 保存到 Artifacts 中。
set_local	设置此运行是否为本地运行。

batch_save_artifacts

将一批文本文件和模型保存到 Artifacts 中。 这比保存单个项目更高效。

batch_save_artifacts(working_directory: Optional[str], input_strs: Dict[str, str], model_outputs: Dict[str, Any], save_as_mlflow: bool = False, mlflow_options: Optional[Dict[str, Any]] = None) -> None

参数

working_directory

必需

用于临时存储的目录。

input_strs

必需

字符串字典。 键是项目名称，值是内容。

model_outputs

必需

模型字典。 键是项目名称，值是模型。

mlflow_options

默认值: False

用于模型保存过程的 MLflow 设置/包装器的字典。

save_as_mlflow

默认值: None

表示是否使用 MLFlow 来保存模型的标志。

get_run

生成运行上下文。

通过 contextlib 进行包装，以将其转换为上下文管理器。 嵌套调用将返回相同的运行上下文。

get_run()

save_model_output

使用此运行上下文将给定的拟合模型保存到给定路径。

save_model_output(fitted_pipeline: Any, remote_path: str, working_dir: str) -> None

参数

fitted_pipeline

必需

要保存的拟合模型

remote_path

必需

要保存到的路径

working_dir

必需

save_onnx_model_output

使用此运行上下文将给定的 onnx 模型保存到给定的远程路径。

save_onnx_model_output(onnx_model: object, remote_path: str, working_dir: str) -> None

参数

onnx_model

必需

要保存的 onnx 模型

remote_path

必需

要保存到的路径

working_dir

必需

save_onnx_model_resource

使用此运行上下文将给定的 onnx 模型资源保存到给定的远程路径。

save_onnx_model_resource(onnx_resource: Dict[Any, Any], remote_path: str, working_dir: str) -> None

参数

onnx_resource

必需

要保存的 onnx 模型资源字典

remote_path

必需

要保存到的路径

working_dir

必需

save_str_output

将 str 文件作为 txt 保存到 Artifacts 中。

save_str_output(input_str: str, remote_path: str, overwrite_mode: bool = False, working_directory: Optional[str] = None) -> None

参数

input_str

必需

输入字符串。

remote_path

必需

Artifacts 中的文件名。

overwrite_mode

默认值: False

working_directory

默认值: None

set_local

设置此运行是否为本地运行。

set_local(local: bool) -> None

参数

local

必需

此运行是否为本地运行

属性

parent_run_id

获取此执行上下文的父运行 ID；如果这是父运行，则获取运行 ID。

返回

父运行 ID

run_id

获取与此运行上下文所包装的运行关联的运行 ID。 假设运行 ID 不可变。

返回

运行 ID