diff options
Diffstat (limited to 'regress_pred.m')
-rw-r--r-- | regress_pred.m | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/regress_pred.m b/regress_pred.m index 2992471..119738c 100644 --- a/regress_pred.m +++ b/regress_pred.m @@ -1,4 +1,4 @@ -function pred = regress_pred(price, quantity) +function pred = regress_pred(price, quantity, prefunc) days = size(price, 1) - size(quantity, 1); product_count = size(price, 2); @@ -8,8 +8,8 @@ function pred = regress_pred(price, quantity) pred = zeros(days, product_count); for i = 1:product_count - mn = regress(price(test_set, i), quantity(test_set, i)); - pred(:, i) = polyval(mn, price(pred_set, i)); + mn = regress(prefunc(price(test_set, i)), quantity(test_set, i)); + pred(:, i) = polyval(mn, prefunc(price(pred_set, i))); end endfunction |