@@ -64,8 +64,8 @@ class PaLM2TextGenerator(base.BaseEstimator):
6464 BQ session to create the model. If None, use the global default session.
6565 connection_name (str or None):
6666 Connection to connect with remote service. str of the format <PROJECT_NUMBER/PROJECT_ID>.<LOCATION>.<CONNECTION_ID>.
67- if None, use default connection in session context. BigQuery DataFrame will try to create the connection and attach
68- permission if the connection isn't fully setup .
67+ If None, use default connection in session context. BigQuery DataFrame will try to create the connection and attach
68+ permission if the connection isn't fully set up .
6969 max_iterations (Optional[int], Default to 300):
7070 The number of steps to run when performing supervised tuning.
7171 """
@@ -191,7 +191,7 @@ def fit(
191191 Training labels.
192192
193193 Returns:
194- PaLM2TextGenerator: Fitted Estimator .
194+ PaLM2TextGenerator: Fitted estimator .
195195 """
196196 X , y = utils .convert_to_dataframe (X , y )
197197
@@ -372,12 +372,12 @@ def to_gbq(self, model_name: str, replace: bool = False) -> PaLM2TextGenerator:
372372
373373 Args:
374374 model_name (str):
375- the name of the model.
375+ The name of the model.
376376 replace (bool, default False):
377377 Determine whether to replace if the model already exists. Default to False.
378378
379379 Returns:
380- PaLM2TextGenerator: saved model."""
380+ PaLM2TextGenerator: Saved model."""
381381
382382 new_model = self ._bqml_model .copy (model_name , replace )
383383 return new_model .session .read_gbq_model (model_name )
@@ -390,16 +390,16 @@ class PaLM2TextEmbeddingGenerator(base.BaseEstimator):
390390 Args:
391391 model_name (str, Default to "textembedding-gecko"):
392392 The model for text embedding. “textembedding-gecko” returns model embeddings for text inputs.
393- "textembedding-gecko-multilingual" returns model embeddings for text inputs which support over 100 languages
393+ "textembedding-gecko-multilingual" returns model embeddings for text inputs which support over 100 languages.
394394 Default to "textembedding-gecko".
395395 version (str or None):
396396 Model version. Accepted values are "001", "002", "003", "latest" etc. Will use the default version if unset.
397397 See https://cloud.google.com/vertex-ai/docs/generative-ai/learn/model-versioning for details.
398398 session (bigframes.Session or None):
399399 BQ session to create the model. If None, use the global default session.
400400 connection_name (str or None):
401- connection to connect with remote service. str of the format <PROJECT_NUMBER/PROJECT_ID>.<LOCATION>.<CONNECTION_ID>.
402- if None, use default connection in session context.
401+ Connection to connect with remote service. str of the format <PROJECT_NUMBER/PROJECT_ID>.<LOCATION>.<CONNECTION_ID>.
402+ If None, use default connection in session context.
403403 """
404404
405405 def __init__ (
@@ -539,12 +539,12 @@ def to_gbq(
539539
540540 Args:
541541 model_name (str):
542- the name of the model.
542+ The name of the model.
543543 replace (bool, default False):
544544 Determine whether to replace if the model already exists. Default to False.
545545
546546 Returns:
547- PaLM2TextEmbeddingGenerator: saved model."""
547+ PaLM2TextEmbeddingGenerator: Saved model."""
548548
549549 new_model = self ._bqml_model .copy (model_name , replace )
550550 return new_model .session .read_gbq_model (model_name )
@@ -565,8 +565,8 @@ class GeminiTextGenerator(base.BaseEstimator):
565565 BQ session to create the model. If None, use the global default session.
566566 connection_name (str or None):
567567 Connection to connect with remote service. str of the format <PROJECT_NUMBER/PROJECT_ID>.<LOCATION>.<CONNECTION_ID>.
568- if None, use default connection in session context. BigQuery DataFrame will try to create the connection and attach
569- permission if the connection isn't fully setup .
568+ If None, use default connection in session context. BigQuery DataFrame will try to create the connection and attach
569+ permission if the connection isn't fully set up .
570570 """
571571
572572 def __init__ (
@@ -719,12 +719,12 @@ def to_gbq(self, model_name: str, replace: bool = False) -> GeminiTextGenerator:
719719
720720 Args:
721721 model_name (str):
722- the name of the model.
722+ The name of the model.
723723 replace (bool, default False):
724724 Determine whether to replace if the model already exists. Default to False.
725725
726726 Returns:
727- GeminiTextGenerator: saved model."""
727+ GeminiTextGenerator: Saved model."""
728728
729729 new_model = self ._bqml_model .copy (model_name , replace )
730730 return new_model .session .read_gbq_model (model_name )
0 commit comments