Problème avec custom loss fonction sur Keras

Bonjour à tous,

je souhaiterais utiliser une fonction d'optimisation pour une réseau de neurone or je ne parviens pas à la faire fonctionner.

Code:

---

1 2 3 4 5 6

def bidule(y_true, y_pred):   product = np.fft.fft(y_true) * np.fft.fft( y_pred).conj() cc = np.fft.fftshift(np.fft.ifft(product)) return np.argmax(cc.real)

---

Lorsque je lance mon réseau j’obtiens le message d'erreur "ValueError: Invalid axis (-1) specified."
Je ne comprend pas si cette erreur vient de la fonction np.fft ou d'ailleurs.
J'ai vu qu'il était aussi possible d'utilise certaines fonctions propres à keras (comme K.mean, K.square) donc je me demande si l'origine du pb ne viendrait pas du fait que j'utilise la fonction fft de numpy et non de keras.

Bonne soirée et merci pour votre aide !

Nicolas