Module imported (1.20.0)

Imported models.

Classes

ONNXModel

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

Imported Open Neural Network Exchange (ONNX) model.

Parameters
NameDescription
model_pathstr

Cloud Storage path that holds the model files.

sessionBigQuery Session

BQ session to create the model.

TensorFlowModel

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

Imported TensorFlow model.

Parameters
NameDescription
model_pathstr

Cloud Storage path that holds the model files.

sessionBigQuery 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
NameDescription
model_pathstr

Cloud Storage path that holds the model files.

inputDict, 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

outputDict, 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

sessionBigQuery Session

BQ session to create the model.