FIX: fix predicted values for both linear and polynomial regressions