class

CrystalML::Clustering::KMeans

Inherits CrystalML::Clusterer / CrystalML::UnsupervisedEstimator / CrystalML::BaseEstimator / Reference / Object

Constructors

new(n_clusters : Int32 = 8, max_iters : Int32 = 300, tolerance : Float64 = 0.0001)
Source

Instance methods

centroids
Source
fit(data : Tensor(Float64, CPU(Float64)))
Source
max_iters
Source
max_iters=(max_iters : Int32)
Source
n_clusters
Source
n_clusters=(n_clusters : Int32)
Source
predict(data : Tensor(Float64, CPU(Float64))) : Array(Int32)
Source
tolerance
Source
tolerance=(tolerance : Float64)
Source