Module imported (0.21.0)

Imported models.

Classes

ONNXModel

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

Imported Open Neural Network Exchange (ONNX) model.

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

TensorFlowModel

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

Imported TensorFlow model.

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

XGBoostModel

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

Imported XGBoost model.

Parameters
Name	Description
`session`	`BigQuery Session` BQ session to create the model
`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
`model_path`	`str` Cloud Storage path that holds the model files.