2 instantiations of MatrixFactorizationPredictionTransformer
Microsoft.ML.Recommender (2)
MatrixFactorizationPredictor.cs (1)
560=> new MatrixFactorizationPredictionTransformer(env, ctx);
MatrixFactorizationTrainer.cs (1)
541return new MatrixFactorizationPredictionTransformer(_host, model, trainData.Schema, MatrixColumnIndexName, MatrixRowIndexName);
25 references to MatrixFactorizationPredictionTransformer
Microsoft.ML.IntegrationTests (1)
Evaluation.cs (1)
249var model = pipeline.Fit(data);
Microsoft.ML.Recommender (13)
MatrixFactorizationPredictor.cs (8)
21[assembly: LoadableClass(typeof(MatrixFactorizationPredictionTransformer), typeof(MatrixFactorizationPredictionTransformer), 22null, typeof(SignatureLoadModel), "", MatrixFactorizationPredictionTransformer.LoaderSignature)] 447: base(Contracts.CheckRef(env, nameof(env)).Register(nameof(MatrixFactorizationPredictionTransformer)), model, trainSchema) 479/// The counter constructor of re-creating <see cref="MatrixFactorizationPredictionTransformer"/> from the context where 483: base(Contracts.CheckRef(host, nameof(host)).Register(nameof(MatrixFactorizationPredictionTransformer)), ctx) 557loaderAssemblyName: typeof(MatrixFactorizationPredictionTransformer).Assembly.FullName); 559internal static MatrixFactorizationPredictionTransformer Create(IHostEnvironment env, ModelLoadContext ctx)
MatrixFactorizationTrainer.cs (5)
114ITrainerEstimator<MatrixFactorizationPredictionTransformer, MatrixFactorizationModelParameters> 521/// Trains a <see cref="MatrixFactorizationTrainer"/> using both training and validation data, returns a <see cref="MatrixFactorizationPredictionTransformer"/>. 525public MatrixFactorizationPredictionTransformer Fit(IDataView trainData, IDataView validationData) 545/// <summary> Trains and returns a <see cref="MatrixFactorizationPredictionTransformer"/>.</summary> 548public MatrixFactorizationPredictionTransformer Fit(IDataView input) => Fit(input, null);
Microsoft.ML.Samples (3)
Dynamic\Trainers\Recommendation\MatrixFactorization.cs (1)
40var model = pipeline.Fit(trainingData);
Dynamic\Trainers\Recommendation\MatrixFactorizationWithOptions.cs (1)
64var model = pipeline.Fit(trainingData);
Dynamic\Trainers\Recommendation\OneClassMatrixFactorizationWithOptions.cs (1)
70var model = pipeline.Fit(dataView);
Microsoft.ML.Tests (8)
TrainerEstimators\MatrixFactorizationTests.cs (8)
87var model = pipeline.Fit(data); 153var modelWithValidation = pipeline.Fit(data, testData); 231var model = pipeline.Fit(dataView); 343var model = pipeline.Fit(dataView); 466var model = pipeline.Fit(dataView); 610var model = pipeline.Fit(dataView); 676var model = pipeline.Fit(dataView); 790var model = pipeline.Fit(dataView);