Add AAAI prediction plot function.