namespace Microsoft.ML.Trainers
{
    public sealed class MatrixFactorizationTrainer : IEstimator<MatrixFactorizationPredictionTransformer>, ITrainerEstimator<MatrixFactorizationPredictionTransformer, MatrixFactorizationModelParameters>
    {
        public sealed class Options
        {
            [SweepableDiscreteParam("K", new[] { 8, 16, 64, 128 }, false)]
            public int ApproximationRank;
        }
    }
}