(Without knowing much about your model), did you try replacing features with outputs in this line and see if that works?
features
outputs