Module imported (1.21.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.