We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent b254091 commit 31e345bCopy full SHA for 31e345b
1 file changed
model/ctr/model_test.go
@@ -262,10 +262,11 @@ func TestFactorizationMachines_WeightedTraining(t *testing.T) {
262
foundPurchase := false
263
for i := 0; i < train.Count(); i++ {
264
w := train.GetWeight(i)
265
- if train.FeedbackTypes[i] == "click" {
+ switch train.FeedbackTypes[i] {
266
+ case "click":
267
assert.Equal(t, float32(1.0), w)
268
foundClick = true
- } else if train.FeedbackTypes[i] == "purchase" {
269
+ case "purchase":
270
assert.Equal(t, float32(5.0), w)
271
foundPurchase = true
272
}
0 commit comments