4 instantiations of FieldAwareFactorizationMachineTrainer
Microsoft.ML.StandardTrainers (4)
FactorizationMachine\FactorizationMachineCatalog.cs (3)
40return new FieldAwareFactorizationMachineTrainer(env, new string[] { featureColumnName }, labelColumnName, exampleWeightColumnName); 63return new FieldAwareFactorizationMachineTrainer(env, featureColumnNames, labelColumnName, exampleWeightColumnName); 82return new FieldAwareFactorizationMachineTrainer(env, options);
FactorizationMachine\FactorizationMachineTrainer.cs (1)
580return TrainerEntryPointsUtils.Train<Options, CommonOutputs.BinaryClassificationOutput>(host, input, () => new FieldAwareFactorizationMachineTrainer(host, input),
34 references to FieldAwareFactorizationMachineTrainer
Microsoft.ML.IntegrationTests (3)
Training.cs (3)
51var ffmTrainer = mlContext.BinaryClassification.Trainers.FieldAwareFactorizationMachine(); 139var trainer = mlContext.BinaryClassification.Trainers.FieldAwareFactorizationMachine( 140new FieldAwareFactorizationMachineTrainer.Options { NumberOfIterations = 100 });
Microsoft.ML.Samples (4)
Dynamic\Trainers\BinaryClassification\FactorizationMachine.cs (1)
35var pipeline = mlContext.BinaryClassification.Trainers
Dynamic\Trainers\BinaryClassification\FieldAwareFactorizationMachine.cs (1)
36var pipeline = mlContext.BinaryClassification.Trainers
Dynamic\Trainers\BinaryClassification\FieldAwareFactorizationMachineWithOptions.cs (2)
31var options = new FieldAwareFactorizationMachineTrainer.Options 52var pipeline = mlContext.BinaryClassification.Trainers
Microsoft.ML.StandardTrainers (23)
FactorizationMachine\FactorizationMachineCatalog.cs (7)
18/// Create <see cref="FieldAwareFactorizationMachineTrainer"/>, which predicts a target using a field-aware factorization machine trained over boolean label data. 33public static FieldAwareFactorizationMachineTrainer FieldAwareFactorizationMachine(this BinaryClassificationCatalog.BinaryClassificationTrainers catalog, 44/// Create <see cref="FieldAwareFactorizationMachineTrainer"/>, which predicts a target using a field-aware factorization machine trained over boolean label data. 56public static FieldAwareFactorizationMachineTrainer FieldAwareFactorizationMachine(this BinaryClassificationCatalog.BinaryClassificationTrainers catalog, 67/// Create <see cref="FieldAwareFactorizationMachineTrainer"/> using advanced options, which predicts a target using a field-aware factorization machine trained over boolean label data. 77public static FieldAwareFactorizationMachineTrainer FieldAwareFactorizationMachine(this BinaryClassificationCatalog.BinaryClassificationTrainers catalog, 78FieldAwareFactorizationMachineTrainer.Options options)
FactorizationMachine\FactorizationMachineTrainer.cs (15)
18[assembly: LoadableClass(FieldAwareFactorizationMachineTrainer.Summary, typeof(FieldAwareFactorizationMachineTrainer), 19typeof(FieldAwareFactorizationMachineTrainer.Options), new[] { typeof(SignatureBinaryClassifierTrainer), typeof(SignatureTrainer) } 20, FieldAwareFactorizationMachineTrainer.UserName, FieldAwareFactorizationMachineTrainer.LoadName, 21FieldAwareFactorizationMachineTrainer.ShortName, DocName = "trainer/FactorizationMachine.md")] 23[assembly: LoadableClass(typeof(void), typeof(FieldAwareFactorizationMachineTrainer), null, typeof(SignatureEntryPointModule), FieldAwareFactorizationMachineTrainer.LoadName)] 92/// <seealso cref="Microsoft.ML.FactorizationMachineExtensions.FieldAwareFactorizationMachine(BinaryClassificationCatalog.BinaryClassificationTrainers, FieldAwareFactorizationMachineTrainer.Options)"/> 93/// <seealso cref="FieldAwareFactorizationMachineTrainer.Options"/> 103/// <see cref="Options"/> for <see cref="FieldAwareFactorizationMachineTrainer"/> as used in 104/// <see cref="Microsoft.ML.FactorizationMachineExtensions.FieldAwareFactorizationMachine(BinaryClassificationCatalog.BinaryClassificationTrainers, FieldAwareFactorizationMachineTrainer.Options)"/>. 217/// Initializes a new instance of <see cref="FieldAwareFactorizationMachineTrainer"/> through the <see cref="Options"/> class. 244/// Initializes a new instance of <see cref="FieldAwareFactorizationMachineTrainer"/>. 585/// Continues the training of a <see cref="FieldAwareFactorizationMachineTrainer"/> using an already trained <paramref name="modelParameters"/> and/or validation data,
FactorizationMachine\FieldAwareFactorizationMachineModelParameters.cs (1)
25/// Model parameters for <see cref="FieldAwareFactorizationMachineTrainer"/>.
Microsoft.ML.Tests (4)
TrainerEstimators\FAFMEstimator.cs (4)
46var ffmArgs = new FieldAwareFactorizationMachineTrainer.Options(); 52var pipeline = mlContext.BinaryClassification.Trainers.FieldAwareFactorizationMachine(ffmArgs); 71var ffmArgs = new FieldAwareFactorizationMachineTrainer.Options 80var est = ML.BinaryClassification.Trainers.FieldAwareFactorizationMachine(ffmArgs);