Struct KMeansModel 

pub struct KMeansModel<T: KMeansElement> { /* private fields */ }

Implementations

impl<T: KMeansElement> KMeansModel<T>

pub fn fit( resources: &CuvsResources, data: &ValidatedDeviceTensor<'_, T>, params: KMeansParams, initial_centroids: Option<DeviceBuffer<T>>, ) -> Result<Self>

pub fn fit_with_initial_centroids( resources: &CuvsResources, data: &ValidatedDeviceTensor<'_, T>, params: KMeansParams, initial_centroids: Option<OwnedMatrix<T>>, ) -> Result<Self>

Fit KMeans with an optional owned row-major centroid relation.

Consumed centroids satisfy the exclusive mutable ownership required by KMeansInit::Array.

pub fn predict_into( &self, data: &ValidatedDeviceTensor<'_, T>, labels: &mut DeviceBuffer<i32>, ) -> Result<KMeansPrediction>

pub fn params(&self) -> &KMeansParams

pub fn fit_result(&self) -> KMeansFit

pub fn centroids(&self) -> &OwnedMatrix<T>

pub fn n_features(&self) -> usize

Trait Implementations

impl<T: Debug + KMeansElement> Debug for KMeansModel<T>

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.