5 references to MatrixFactorization
Microsoft.ML.AutoML (2)
TrainerExtensions\TrainerExtensionCatalog.cs (1)
43{ TrainerName.MatrixFactorization, typeof(MatrixFactorizationExtension) },
TrainerExtensions\TrainerExtensionUtil.cs (1)
384return TrainerName.MatrixFactorization;
Microsoft.ML.CodeGenerator (2)
CodeGenerator\CSharp\CodeGenerator.cs (1)
27private static readonly HashSet<string> _recommendationTrainers = new HashSet<string>() { TrainerName.MatrixFactorization.ToString() };
CodeGenerator\CSharp\TrainerGeneratorFactory.cs (1)
70case TrainerName.MatrixFactorization:
Microsoft.ML.CodeGenerator.Tests (1)
ApprovalTests\ConsoleCodeGeneratorTests.cs (1)
560var matrixPipelineNode = new PipelineNode(nameof(TrainerName.MatrixFactorization), PipelineNodeType.Trainer, "Features", "Score", hyperParam);