Uses 'beta' as key for coef_img_ instead of None in case of a regression