2 instantiations of FieldAwareFactorizationMachinePredictionTransformer
Microsoft.ML.StandardTrainers (2)
FactorizationMachine\FactorizationMachineTrainer.cs (1)
611
return new
FieldAwareFactorizationMachinePredictionTransformer
(_host, model, trainData.Schema, FeatureColumns.Select(x => x.Name).ToArray());
FactorizationMachine\FieldAwareFactorizationMachineModelParameters.cs (1)
443
=> new
FieldAwareFactorizationMachinePredictionTransformer
(env, ctx);
22 references to FieldAwareFactorizationMachinePredictionTransformer
Microsoft.ML.IntegrationTests (4)
Training.cs (4)
61
var
ffm = ffmTrainer.Fit(featurizedTrain);
147
var
firstModel = trainer.Fit(featurizedData);
151
var
firstModelPrime = trainer.Fit(featurizedData);
155
var
secondModel = trainer.Fit(featurizedData, modelParameters: firstModel.Model);
Microsoft.ML.Samples (3)
Dynamic\Trainers\BinaryClassification\FactorizationMachine.cs (1)
39
var
model = pipeline.Fit(trainingData);
Dynamic\Trainers\BinaryClassification\FieldAwareFactorizationMachine.cs (1)
45
var
model = pipeline.Fit(trainingData);
Dynamic\Trainers\BinaryClassification\FieldAwareFactorizationMachineWithOptions.cs (1)
56
var
model = pipeline.Fit(trainingData);
Microsoft.ML.StandardTrainers (12)
FactorizationMachine\FactorizationMachineTrainer.cs (5)
95
IEstimator<
FieldAwareFactorizationMachinePredictionTransformer
>
586
/// and returns a <see cref="
FieldAwareFactorizationMachinePredictionTransformer
"/>.
588
public
FieldAwareFactorizationMachinePredictionTransformer
Fit(IDataView trainData,
614
/// <summary> Trains and returns a <see cref="
FieldAwareFactorizationMachinePredictionTransformer
"/>.</summary>
615
public
FieldAwareFactorizationMachinePredictionTransformer
Fit(IDataView input) => Fit(input, null, null);
FactorizationMachine\FieldAwareFactorizationMachineModelParameters.cs (7)
19
[assembly: LoadableClass(typeof(
FieldAwareFactorizationMachinePredictionTransformer
), typeof(
FieldAwareFactorizationMachinePredictionTransformer
), null, typeof(SignatureLoadModel),
20
"",
FieldAwareFactorizationMachinePredictionTransformer
.LoaderSignature)]
306
: base(Contracts.CheckRef(host, nameof(host)).Register(nameof(
FieldAwareFactorizationMachinePredictionTransformer
)), model, trainSchema)
333
: base(Contracts.CheckRef(host, nameof(host)).Register(nameof(
FieldAwareFactorizationMachinePredictionTransformer
)), ctx)
439
loaderAssemblyName: typeof(
FieldAwareFactorizationMachinePredictionTransformer
).Assembly.FullName);
442
internal static
FieldAwareFactorizationMachinePredictionTransformer
Create(IHostEnvironment env, ModelLoadContext ctx)
Microsoft.ML.Tests (3)
TrainerEstimators\FAFMEstimator.cs (3)
54
var
model = pipeline.Fit(dataView);
83
var
model = est.Fit(data);
84
var
anotherModel = est.Fit(data, data, model.Model);