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