bigframes.ml.cluster.KMeans.fit

KMeans.fit(X: DataFrame | Series | DataFrame | Series, y: DataFrame | Series | DataFrame | Series | None = None) → _T

Compute k-means clustering.

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:

Fitted estimator.

Return type:

KMeans