diff --git a/robot.py b/robot.py index 5ef9e0d..afb02cc 100644 --- a/robot.py +++ b/robot.py @@ -137,7 +137,7 @@ figNum = 1 if len(show_symbols) > 0: for symbol in show_symbols: if symbol in symbols: - title = Stock(symbol, symbols[symbol]) + title = Stock(params, symbol, symbols[symbol]) title.fetch() title.statistics() ind = title.analyze(buy_callback, range_days=params['show_range_days']) diff --git a/stock.py b/stock.py index f03dc3c..3d30cd8 100644 --- a/stock.py +++ b/stock.py @@ -100,10 +100,11 @@ class Stock(object): spl = UnivariateSpline(x=x_r, y=y_r, k=5) spl.set_smoothing_factor(0.25) spl_d = spl.derivative() - spl_dd = spl_d.derivative() + spl_dd = spl.derivative().derivative() yf = spl(x_r) yf_d = spl_d(x_r) - yf_dd = spl_dd(x_r) + x_r2 = np.linspace(-1, N-1, N) + yf_dd = spl_dd(x_r2) self.data['macd_f'] = np.transpose(yf) self.data['macd_fd'] = np.transpose(yf_d)