1 write to _latentDimAligned
Microsoft.ML.StandardTrainers (1)
FactorizationMachine\FactorizationMachineTrainer.cs (1)
286
_latentDimAligned
= FieldAwareFactorizationMachineUtils.GetAlignedVectorLength(_latentDim);
13 references to _latentDimAligned
Microsoft.ML.StandardTrainers (13)
FactorizationMachine\FactorizationMachineTrainer.cs (13)
301
latentWeightsAligned = new AlignedArray(featureCount * fieldCount *
_latentDimAligned
, 16);
303
latentAccumulatedSquaredGradsAligned = new AlignedArray(featureCount * fieldCount *
_latentDimAligned
, 16);
314
int vBias = j * fieldCount *
_latentDimAligned
+ f *
_latentDimAligned
;
315
for (int k = 0; k <
_latentDimAligned
; k++)
335
int vBias = j * fieldCount *
_latentDimAligned
+ f *
_latentDimAligned
;
336
for (int k = 0; k <
_latentDimAligned
; k++)
431
ch.Check(checked(totalFeatureCount * fieldCount *
_latentDimAligned
) <= Utils.ArrayMaxSize, "Latent dimension or the number of fields too large");
462
var latentSum = new AlignedArray(fieldCount * fieldCount *
_latentDimAligned
, 16);
520
FieldAwareFactorizationMachineInterface.CalculateIntermediateVariables(fieldCount,
_latentDimAligned
, count,
525
FieldAwareFactorizationMachineInterface.CalculateGradientAndUpdate(_lambdaLinear, _lambdaLatent, _learningRate, fieldCount,
_latentDimAligned
, weight, count,
539
validLoss = CalculateAvgLoss(ch, validData, _norm, linearWeights, latentWeightsAligned,
_latentDimAligned
, latentSum,