Add output variable NN and SAE into predict_SAE.
Tune predict_SAE parameters.