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.

Methods

__init__(model_path, *[, session])

get_params([deep])

Get parameters for this estimator.

predict(X)

Predict the result from input DataFrame.

register([vertex_ai_model_id])

Register the model to Vertex AI.

to_gbq(model_name[, replace])

Save the model to BigQuery.