Using AI for STP