visualize_shape_model now supports n_params list