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