Hi, thanks for the excellent work! I wonder if the methods could be applied to regression tasks as well?