bigframes.ml.imported.TensorFlowModel#

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

Imported TensorFlow 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. 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) → TensorFlowModel[source]#

Save the model to BigQuery.

Parameters:

• model_name (str) The name of the model.

• replace (bool, default False) Default to False.

Returns:

Saved model.

Return type:

TensorFlowModel