# This contains params to be used by the stages to train or predict