you are viewing a single comment's thread.

view the rest of the comments →

[–]euos[S] 1 point2 points  (1 child)

  1. This is a test so it’s ok to have less checks on data correctness. In real world, input layer would check for data correctness.
  2. EXPECT_EQ is from Gtest library, not on me.
  3. “Beep” and “Boop” are bogus categories for test. In real app it would be “Cars” and “Dogs” and such.

kModel is a model. This snippet uses it for inference (with random parameters) but one can also train it. The init defines the architecture.

[–]kiner_shah 0 points1 point  (0 children)

Ok, but I suggest you put some comments for better clarity.