VertexAI 的 class AutoMLtabularTrainingJob 无法识别文档中列出的参数

Question

我正在尝试使用 VertexAI class AutoMLtabularTrainingJob 创建预测模型，但 documentation.

中列出的两个参数出现问题

首先，参数column_specs在文档中是字典，参数export_evaluated_data_items是bool。

我创建了下面的函数，并在循环中调用了它。

def create_training_pipeline_tabular_regression_sample(
display_name:str,
dataset_id:int,
column_specs:dict,
target_column:str = None,
optimization_prediction_type:str = 'regression',
optimization_objective:str = 'minimize-rmse',
model_display_name:str = None,
budget_milli_node_hours:int = 1000,
disable_early_stopping:bool = False,
export_evaluated_data:bool = True,
sync:bool = True,
**kwargs
):

tabular_regression_job = aiplatform.AutoMLTabularTrainingJob(
    display_name=display_name,
    column_specs=column_specs,
    optimization_prediction_type=optimization_prediction_type,
    optimization_objective=optimization_objective
)

my_tabular_dataset = aiplatform.TabularDataset(dataset_id)

model = tabular_regression_job.run(
    dataset=my_tabular_dataset,
    target_column=target_column,
    budget_milli_node_hours=budget_milli_node_hours,
    model_display_name=model_display_name,
    disable_early_stopping=disable_early_stopping,
    export_evaluated_data_items=True,
    sync=sync,
    **kwargs
)

model.wait()

print(model.display_name)
print(model.resource_name)
print(model.uri)
return model

错误是 class 不接受这些参数。错误信息：

    ---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
/tmp/ipykernel_118/330955058.py in <module>
     60                     optimization_objective=optimization,
     61                     budget_milli_node_hours= BUDGET_MILLI_NODE_HOURS,
---> 62                     export_evaluated_data_items_bigquery_destination_uri=export_evaluated_data_items_bigquery_destination_uri
     63                 )
     64 

/tmp/ipykernel_118/2971171495.py in create_training_pipeline_tabular_regression_sample(display_name, dataset_id, target_column, optimization_prediction_type, optimization_objective, model_display_name, budget_milli_node_hours, disable_early_stopping, export_evaluated_data, sync, **kwargs)
     31         export_evaluated_data_items=True,
     32         sync=sync,
---> 33         **kwargs
     34     )
     35 

TypeError: run() got an unexpected keyword argument 'export_evaluated_data_items'

有谁知道文档更新了吗？在页面的页脚中，更新日期是最近的，但这些错误让我产生了疑问。文档中还有其他信息与 API 的使用不符。

Answer 1

当我降级到 google-cloud-aiplatform 版本 0.7.1 时，我能够重现你的错误。要解决此问题，您必须使用以下命令将您的版本更新到最新的 google-cloud-aiplatform 软件包。

pip install google-cloud-aiplatform --upgrade

您现在将拥有 google-cloud-aiplatform 版本 1.13.1。

升级到最新版本后，您现在可以继续并完成培训。

VertexAI 的 class AutoMLtabularTrainingJob 无法识别文档中列出的参数

VertexAI's class AutoMLtabularTrainingJob doesn't recognize parameters listed in the documentation

python

machine-learning

google-cloud-platform

google-cloud-vertex-ai