Generalization of WNN

[logbycat]

Hello, regarding the part of the wavelet neural network in your code, I found that the generalization of the model is very low. Would you like to give me some advice? thank you!