fitted function

extract the predictions