Module imported (1.14.0)

Imported models.

Classes

ONNXModel

ONNXModel(
    model_path: str, *, session: typing.Optional[bigframes.session.Session] = None
)

Imported Open Neural Network Exchange (ONNX) model.

Parameters
Name	Description
`model_path`	`str` Cloud Storage path that holds the model files.
`session`	`BigQuery Session` BQ session to create the model.

TensorFlowModel

TensorFlowModel(
    model_path: str, *, session: typing.Optional[bigframes.session.Session] = None
)

Imported TensorFlow model.

Parameters
Name	Description
`model_path`	`str` Cloud Storage path that holds the model files.
`session`	`BigQuery Session` BQ session to create the model.

XGBoostModel

XGBoostModel(
    model_path: str,
    *,
    input: typing.Mapping[str, str] = {},
    output: typing.Mapping[str, str] = {},
    session: typing.Optional[bigframes.session.Session] = None
)

Imported XGBoost model.

Parameters
Name	Description
`model_path`	`str` Cloud Storage path that holds the model files.
`input`	`Dict, default None` Specify the model input schema information when you create the XGBoost model. The input should be the format of {field_name: field_type}. Input is optional only if feature_names and feature_types are both specified in the model file. Supported types are "bool", "string", "int64", "float64", "array
`output`	`Dict, default None` Specify the model output schema information when you create the XGBoost model. The input should be the format of {field_name: field_type}. Output is optional only if feature_names and feature_types are both specified in the model file. Supported types are "bool", "string", "int64", "float64", "array
`session`	`BigQuery Session` BQ session to create the model.