diff options
Diffstat (limited to 'quad_regress_pred.m')
-rw-r--r-- | quad_regress_pred.m | 9 |
1 files changed, 3 insertions, 6 deletions
diff --git a/quad_regress_pred.m b/quad_regress_pred.m index aef8822..d9e3b6e 100644 --- a/quad_regress_pred.m +++ b/quad_regress_pred.m @@ -8,16 +8,13 @@ function pred = quad_regress_pred(price, quantity) pred = zeros(days, product_count); for i = 1:product_count - [a,b,c] = regress2(price(test_set, i), quantity(test_set, i)); - pred(:, i) = a .* price(pred_set, i).^2 + b * price(pred_set, i) + c; + abc = regress2(price(test_set, i), quantity(test_set, i)); + pred(:, i) = polyval(abc, price(pred_set, i)); end endfunction -function [a,b,c] = regress2(x, y) +function abc = regress2(x, y) abc = [ x.^2 x ones(length(x), 1) ] \ y; - a = abc(1); - b = abc(2); - c = abc(3); endfunction |