ADFS is a good choice when authentication is required to be only on-prem.
For all other cases the use of PTA or PHS would be more beneficial due to PHS aids with better availability and has no reliability on on-premises elements.
Migrating from ADFS to Azure AD is an important step if your business wants to become fully cloud based, but it important to be prepared for some challenges along the way.