Coords and dims as just labels for your shape, so they can’t solve any shape by themselves (although it often makes it easier to reason about).
You may want to see if bambi could help you create the right regression model. They have support for one kind of multinomial regression: Add multinomial family by tomicapretto · Pull Request #490 · bambinos/bambi · GitHub