We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 9f354f5 commit 82ffe2aCopy full SHA for 82ffe2a
1 file changed
model/ctr/model_test.go
@@ -271,10 +271,11 @@ func TestFactorizationMachines_WeightedTraining(t *testing.T) {
271
foundPurchase := false
272
for i := 0; i < train.Count(); i++ {
273
w := train.GetWeight(i)
274
- if train.FeedbackTypes[i] == "click" {
+ switch train.FeedbackTypes[i] {
275
+ case "click":
276
assert.Equal(t, float32(1.0), w)
277
foundClick = true
- } else if train.FeedbackTypes[i] == "purchase" {
278
+ case "purchase":
279
assert.Equal(t, float32(5.0), w)
280
foundPurchase = true
281
}
0 commit comments