From 020f82903d960f379a3b127c0f46d6bf1106cb26 Mon Sep 17 00:00:00 2001 From: Benjamin Franzke Date: Sat, 12 May 2012 15:59:20 +0200 Subject: regress_pred: Add degree param, to fold in quad regression --- regress_pred.m | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'regress_pred.m') 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 -- cgit