vertexai's class automltabulartrainingjob do do not d识别文档中列出的参数

Question

我正在尝试使用vertexai类automltabulartrainingjob创建一个预测模型，并且我在文档。

首先，参数column_specs是文档中的字典，而参数export_evaLET_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

错误是类不接受这些参数。错误消息：

    ---------------------------------------------------------------------------
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。

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