2 instantiations of FieldAwareFactorizationMachineModelParameters
Microsoft.ML.StandardTrainers (2)
FactorizationMachine\FactorizationMachineTrainer.cs (1)
550return new FieldAwareFactorizationMachineModelParameters(_host, _norm, fieldCount, totalFeatureCount, _latentDim, linearWeights, latentWeightsAligned);
FactorizationMachine\FieldAwareFactorizationMachineModelParameters.cs (1)
183return new FieldAwareFactorizationMachineModelParameters(env, ctx);
20 references to FieldAwareFactorizationMachineModelParameters
Microsoft.ML.StandardTrainers (20)
FactorizationMachine\FactorizationMachineTrainer.cs (12)
94public sealed class FieldAwareFactorizationMachineTrainer : ITrainer<FieldAwareFactorizationMachineModelParameters>, 297private void InitializeTrainingState(int fieldCount, int featureCount, FieldAwareFactorizationMachineModelParameters predictor, out float[] linearWeights, 408private FieldAwareFactorizationMachineModelParameters TrainCore(IChannel ch, IProgressChannel pch, RoleMappedData data, 409RoleMappedData validData = null, FieldAwareFactorizationMachineModelParameters predictor = null) 553private FieldAwareFactorizationMachineModelParameters Train(TrainContext context) 556var initPredictor = context.InitialPredictor as FieldAwareFactorizationMachineModelParameters; 558"Initial predictor should have been " + nameof(FieldAwareFactorizationMachineModelParameters)); 568FieldAwareFactorizationMachineModelParameters ITrainer<FieldAwareFactorizationMachineModelParameters>.Train(TrainContext context) => Train(context); 589IDataView validationData = null, FieldAwareFactorizationMachineModelParameters modelParameters = null) 591FieldAwareFactorizationMachineModelParameters model = null;
FactorizationMachine\FieldAwareFactorizationMachineModelParameters.cs (6)
17[assembly: LoadableClass(typeof(FieldAwareFactorizationMachineModelParameters), null, typeof(SignatureLoadModel), "Field Aware Factorization Machine", FieldAwareFactorizationMachineModelParameters.LoaderSignature)] 60loaderAssemblyName: typeof(FieldAwareFactorizationMachineModelParameters).Assembly.FullName); 178internal static FieldAwareFactorizationMachineModelParameters Create(IHostEnvironment env, ModelLoadContext ctx) 287public sealed class FieldAwareFactorizationMachinePredictionTransformer : PredictionTransformerBase<FieldAwareFactorizationMachineModelParameters> 304internal FieldAwareFactorizationMachinePredictionTransformer(IHostEnvironment host, FieldAwareFactorizationMachineModelParameters model, DataViewSchema trainSchema,
FactorizationMachine\FieldAwareFactorizationMachineUtils.cs (2)
59private readonly FieldAwareFactorizationMachineModelParameters _pred; 74DataViewSchema outputSchema, FieldAwareFactorizationMachineModelParameters pred)