summaryrefslogtreecommitdiff
path: root/regress_pred.m
diff options
context:
space:
mode:
authorBenjamin Franzke <benjaminfranzke@googlemail.com>2012-05-12 15:59:20 +0200
committerBenjamin Franzke <benjaminfranzke@googlemail.com>2012-05-12 15:59:20 +0200
commit020f82903d960f379a3b127c0f46d6bf1106cb26 (patch)
tree8de5740b2d08d8e4063d639bbebf1a6f6c2de5ec /regress_pred.m
parentcc8b965d52177eb45185d0b1ead65b55f113a687 (diff)
downloaddmc-020f82903d960f379a3b127c0f46d6bf1106cb26.tar.gz
dmc-020f82903d960f379a3b127c0f46d6bf1106cb26.tar.bz2
dmc-020f82903d960f379a3b127c0f46d6bf1106cb26.zip
regress_pred: Add degree param, to fold in quad regression
Diffstat (limited to 'regress_pred.m')
-rw-r--r--regress_pred.m4
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