I\'m currently trying to replicate the following classification module from this paper (https://ieeexplore.ieee.org/stamp/stamp.jsp?arnumber=9206937&casa_token=UW7CpTT3g