AI approach could predict treatment strategy for arrhythmia