DOC shape=(model.size_psi,), not (model.psi,)
rename [batch_]psi methods to [batch_]joint_features

What to do with size_psi?