Planar data classification input/output size question

My intuition on those size is 2,1
Each input takes 2 params, and the output is 1 param

why are the values showing 5,2

That is just a test case for the initialization function. We are trying to write general code here, which is reusable with any sized inputs and is not “hard-coded” to the specific input data we have here. That will be a theme here anytime we write functions. The point of functions is that they are modular and reusable.