Int Y_predict[N,K];

The first one is correct

There’s a similar discussion in How to declare a integer-valued matrix? - #4 by Xavier_La_Rochelle