Model generalization

shape