Add ML based optimization to PCI opt