顶点AI提供了一个非常有趣的模型注册表,允许您存储所有经过培训的模型,并跟踪它们的所有版本。
但是,我无法使用Python创建相同模型的新版本。具体而言,我有一个顶点AI管道,它执行: 1)数据预处理,2)特征工程,3)特征存储创建,最后,4)用AutoML表格训练一个模型。
专用于第4点的管道组件的代码是:
automl_training_electric_op = gcc_aip.AutoMLTabularTrainingJobRunOp(
project=project_bq,
model_display_name="pred-model",
display_name="pred-model",
optimization_prediction_type="classification",
optimization_objective="maximize-au-prc",
budget_milli_node_hours=1000,
dataset=comp5a.outputs["dataset"],
target_column="fault",
location=location
)
在Google文档中,我没有发现任何可以帮助我创建新版本的"pred- model ",事实上,每当我运行管道时,顶点AI都会创建一个同名的新模型。
我希望在每次培训中,AutoML都会创建一个相同模型的新版本。例如,v1,v2,v3。
这里是当前的情况,在这种情况下,相同的模型被复制而不是版本化:
发布于 2022-09-02 15:06:16
AutoMLTabularTrainingJobRunOp上一次更新是2022年2月2日。同时,2022年6月28日,对模型版本控制的支持是对顶点Python的已添加。因此,问题是需要更新AutoMLTabularTrainingJobRunOp
,使其与SDK当前提供的内容保持一致,这可能需要提交一个GitHub问题。
同时,您可以使用Python,它支持指定版本。这可以封装在一个基于函数的组件中,用于KFP管道中。
更新(2022-10-05):最新版本中的组件现在支持与模型版本控制相关的新参数:链接
https://stackoverflow.com/questions/73573487
复制相似问题