bigframes.ml.imported.ONNXModel#

class bigframes.ml.imported.ONNXModel(model_path: str, *, session: Session | None = None)[source]#

Imported Open Neural Network Exchange (ONNX) model.

Parameters:

  • model_path (str) – Cloud Storage path that holds the model files.

  • session (BigQuery Session) – BQ session to create the model.

predict(X: DataFrame | Series | DataFrame | Series) → DataFrame[source]#

Predict the result from input DataFrame.

Parameters:

X (bigframes.dataframe.DataFrame or bigframes.series.Series or pandas.core.frame.DataFrame or pandas.core.series.Series) – Input DataFrame or Series. Schema is defined by the model.

Returns:

Output DataFrame, schema is defined by the model.

Return type:

bigframes.dataframe.DataFrame

to_gbq(model_name: str, replace: bool = False) → ONNXModel[source]#

Save the model to BigQuery.

Parameters:

  • model_name (str) – The name of the model.

  • replace (bool, default False) – Determine whether to replace if the model already exists. Default to False.

Returns:

Saved model.

Return type:

ONNXModel