I think you can only understand certain aspect/behaviour of EDFA through modeling in COMSOL as a device. As such EDFA is a system which is very difficult to realize in COMSOL as per my understanding. 

Hi

I think its better you use Lumerical instead of comsol.