Hi Frank,
Thanks for the detailed answer, I will update after I’ll give it a try,
I think that cross entropy loss is a good idea

I thought about this issue regarding argmin()…

some context:
I’m trying to build a network which will predict an inverse function for a channel distortion,
I simulate a signal and pass it trough a random FIR filter (channel distortion function), later I pass this data and the actual data to the network. The network outputs filter coeffs that are suppose to inverse the FIR response.
*the filter is lfilter from torch audio
The argmin() suppose to be used as a detector, I calculate the distance between the reconstructed signal and the initial code signal, argmin is the detected code.
example: symMap = pd.Series(data=[-1, 1j, -1j, 1], index=[0, 1, 2, 3])
than if the reconstructed signal is -0.5 the detected data will be 0.
Thanks,
Ilya