TabularCloudPredictor.predict#

TabularCloudPredictor.predict(test_data: Union[str, DataFrame], test_data_image_column: Optional[str] = None, predictor_path: Optional[str] = None, framework_version: str = 'latest', job_name: Optional[str] = None, instance_type: str = 'ml.m5.2xlarge', instance_count: int = 1, custom_image_uri: Optional[str] = None, wait: bool = True, backend_kwargs: Optional[Dict] = None) → Optional[Series]#

批量推断。当延迟不是主要关注点时，批量转换功能可能更容易、更具可扩展性且更适合。如果您想最小化延迟，请改用 predict_real_time()。

参数

  • test_data (Union(str, pandas.DataFrame)) – 要进行推断的测试数据。可以是 pandas.DataFrame 或本地 csv 文件路径。

  • test_data_image_column (str, default = None) – 如果 test_data 涉及图像模态，您必须指定对应图像路径的列名。该路径必须是绝对路径。

  • predictor_path (str) – 要用于预测的预测器 tarball 路径。路径可以是本地路径或 S3 位置。如果为 None，则将使用通过 fit() 训练的最新预测器。

  • framework_version (str, default = latest) – AutoGluon 的推断容器版本。如果为 latest，将使用最新的可用容器版本。如果提供了特定版本，将使用该版本。如果设置了 custom_image_uri，此参数将被忽略。

  • job_name (str, default = None) – 启动的训练作业名称。如果为 None，CloudPredictor 将创建一个以 ag-cloudpredictor 为前缀的名称。

  • instance_count (int, default = 1,) – 用于批量转换的实例数量。

  • instance_type (str, default = 'ml.m5.2xlarge') – 用于批量转换的实例类型。

  • wait (bool, default = True) – 是否等待批量转换完成。请注意，函数不会立即返回，因为在转换之前需要进行一些准备工作。

  • backend_kwargs (dict, default = None) –

    需要传递到底层后端的所有额外参数。对于 SageMaker 后端，有效键包括：

    1. download: bool, default = True

      是否将批量转换结果下载到磁盘并在批量转换完成后加载。如果 wait 为 False，则此参数将被忽略。

    2. persist: bool, default = True

      是否将下载的批量转换结果持久化到磁盘。如果 download 为 False，则此参数将被忽略。

    3. save_path: str, default = None,

      保存下载结果的路径。如果 download 为 False，则此参数将被忽略。如果为 None，CloudPredictor 将创建一个路径。如果 persist 为 False，文件将首先下载到此路径然后被移除。

    4. model_kwargs: dict, default = dict()

      初始化 Sagemaker Model 所需的任何额外参数。所有选项请参考 https://sagemaker.readthedocs.io/en/stable/api/inference/model.html#model

    5. transformer_kwargs: dict

      需要传递给 Transformer 的任何额外参数。所有选项请参考 https://sagemaker.readthedocs.io/en/stable/api/inference/transformer.html#sagemaker.transformer.Transformer。

    6. transform_kwargs

      需要传递给 transform 的任何额外参数。所有选项请参考 https://sagemaker.readthedocs.io/en/stable/api/inference/transformer.html#sagemaker.transformer.Transformer.transform。

返回值

如果 download 为 True，则返回 Series 中的预测结果；如果 download 为 False，则返回 None。

返回类型

Optional Pandas.Series