Learning high-order interactions for polygenic risk prediction