namespace Microsoft.ML.Trainers
{
    public class KMeansTrainer : TrainerEstimatorBase<ClusteringPredictionTransformer<KMeansModelParameters>, KMeansModelParameters>
    {
        public sealed class Options : UnsupervisedTrainerInputBaseWithWeight
        {
            [SweepableDiscreteParam("K", new[] { 5, 10, 20, 40 }, false)]
            public int NumberOfClusters;
        }
    }
}