3 writes to LatentDimension
Microsoft.ML.StandardTrainers (3)
FactorizationMachine\FieldAwareFactorizationMachineModelParameters.cs (3)
89
LatentDimension
= latentDim;
124
LatentDimension
= latentDim;
158
LatentDimension
= latentDim;
19 references to LatentDimension
Microsoft.ML.StandardTrainers (19)
FactorizationMachine\FactorizationMachineTrainer.cs (1)
435
ch.Check(predictor.
LatentDimension
== _latentDim, "Input model's latent dimension mismatches trainer's");
FactorizationMachine\FieldAwareFactorizationMachineModelParameters.cs (18)
73
/// and each latent vector contains <see cref="
LatentDimension
"/> values. In the f-th field, the j-th feature's latent vector, `v_{j, f}` in the doc
98
int index = j * FieldCount *
LatentDimension
+ f *
LatentDimension
;
102
if (k <
LatentDimension
)
165
int vBias = j * FieldCount *
LatentDimension
+ f *
LatentDimension
;
169
if (k <
LatentDimension
)
204
Host.Assert(
LatentDimension
> 0);
211
ctx.Writer.Write(
LatentDimension
);
213
float[] latentWeights = new float[FeatureCount * FieldCount *
LatentDimension
];
218
int vBias = j * FieldCount *
LatentDimension
+ f *
LatentDimension
;
220
for (int k = 0; k <
LatentDimension
; k++)
263
/// and each latent vector contains <see cref="
LatentDimension
"/> values. In the f-th field, the j-th feature's latent vector, `v_{j, f}` in the doc
270
var latentWeights = new float[FeatureCount * FieldCount *
LatentDimension
];
275
int index = j * FieldCount *
LatentDimension
+ f *
LatentDimension
;
277
for (int k = 0; k <
LatentDimension
; k++)