@heusch,
Following our today's discussion, here is the update to the autoencoder class allowing to avoid prediction_function in the configs. Please merge if fine with you. Thanks!
prediction_function