Building a 9-ball AI player: Candidate generation for direct cut shots [P]

I'm building a 9-ball-player to help with pattern play. There are many ways to make the next ball, and sometimes in than one obvious pocket. Which should should you choose depends on probability of making that shot AND ending up in a favorable spot for the next shot, that is also amenable to getting good position for the shot after. To that end, I have built the following components: A transformer based model that learns p(win) given a table layout. Candidate shot generator that includes cut shots, bank shots, kick shots, caroms and combination shots as well as safeties.