bigframes.ml.cluster.KMeans.fit_predict#
- KMeans.fit_predict(X: DataFrame | Series | DataFrame | Series, y: DataFrame | Series | DataFrame | Series | None = None) _T#
Compute cluster centers and predict cluster index for each sample.
Convenience method; equivalent to calling fit(X) followed by predict(X).
- Parameters:
X (bigframes.dataframe.DataFrame or bigframes.series.Series or pandas.core.frame.DataFrame or pandas.core.series.Series) – DataFrame of shape (n_samples, n_features). Training data.
y (default None) – Not used, present here for API consistency by convention.
- Returns:
DataFrame of shape (n_samples, n_input_columns + n_prediction_columns). Returns predicted labels.
- Return type: