Locally generalised multi-agent reinforcement learning for demand and capacity balancing with customised neural networks