3 writes to FeatureCount
Microsoft.ML.StandardTrainers (3)
FactorizationMachine\FieldAwareFactorizationMachineModelParameters.cs (3)
88FeatureCount = featureCount; 123FeatureCount = featureCount; 157FeatureCount = featureCount;
16 references to FeatureCount
Microsoft.ML.StandardTrainers (16)
FactorizationMachine\FactorizationMachineTrainer.cs (1)
434ch.Check(predictor.FeatureCount == totalFeatureCount, "Input model's feature count mismatches training feature count");
FactorizationMachine\FieldAwareFactorizationMachineModelParameters.cs (12)
92_latentWeightsAligned = new AlignedArray(FeatureCount * FieldCount * LatentDimAligned, 16); 94for (int j = 0; j < FeatureCount; j++) 160_latentWeightsAligned = new AlignedArray(FeatureCount * FieldCount * LatentDimAligned, 16); 161for (int j = 0; j < FeatureCount; j++) 203Host.Assert(FeatureCount > 0); 205Host.Assert(Utils.Size(_linearWeights) == FeatureCount); 206Host.Assert(_latentWeightsAligned.Size == FeatureCount * FieldCount * LatentDimAligned); 210ctx.Writer.Write(FeatureCount); 213float[] latentWeights = new float[FeatureCount * FieldCount * LatentDimension]; 214for (int j = 0; j < FeatureCount; j++) 270var latentWeights = new float[FeatureCount * FieldCount * LatentDimension]; 271for (int j = 0; j < FeatureCount; j++)
FactorizationMachine\FieldAwareFactorizationMachineUtils.cs (3)
103var featureFieldBuffer = new int[_pred.FeatureCount]; 104var featureIndexBuffer = new int[_pred.FeatureCount]; 105var featureValueBuffer = new float[_pred.FeatureCount];