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