bigframes.ml.llm.MultimodalEmbeddingGenerator.predict#
- MultimodalEmbeddingGenerator.predict(X: DataFrame | Series | DataFrame | Series, *, max_retries: int = 0) 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, can contain one or more columns. If multiple columns are in the DataFrame, it must contain a “content” column for prediction. The content column must be of string type or BigFrames Blob of image or video.
max_retries (int, default 0) – Max number of retries if the prediction for any rows failed. Each try needs to make progress (i.e. has successfully predicted rows) to continue the retry. Each retry will append newly succeeded rows. When the max retries are reached, the remaining rows (the ones without successful predictions) will be appended to the end of the result.
- Returns:
DataFrame of shape (n_samples, n_input_columns + n_prediction_columns). Returns predicted values.
- Return type: