From cc8b965d52177eb45185d0b1ead65b55f113a687 Mon Sep 17 00:00:00 2001 From: Benjamin Franzke Date: Sat, 12 May 2012 15:08:13 +0200 Subject: Fold log_regress_pred into regress_pred By adding a pre-filter function pointer. --- regress_pred.m | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'regress_pred.m') 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 -- cgit