Partial pooled model with complex categories

Yep, the two should be equivalent, but using matrix multiplication would be more flexible and easier to extend in most case.