Make dense columns durring prediction as a naive correctness implementation