Hello Divya,
you may use the Python Component in OptiSystem and include your ML code in it. You may find references on how to use the component and signal structure in Example Library of OptiSystem located at:
C:\Users\xxxxxxx\Documents\OptiSystem 16.0 Samples\PythonScripts
Also, you may find some examples on COADM in OptiSystem example at:
C:\Users\xxxxxxx\Documents\OptiSystem 16.0 Samples\WDM systems\
Regards,
Ahmad