jerry/OpenBB
1
0
Fork 0
mirror of https://github.com/OpenBB-finance/OpenBB.git synced 2026-10-27 12:40:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Updated vsurf and forecast API call (#2843)

Browse Source 
* Updated vsurf API call

* Added fixes to forecast dashboard

Co-authored-by: James Maslek <jmaslek11@gmail.com>
This commit is contained in:
Colin Delahunty
2022-10-14 18:44:55 +01:00
committed by GitHub
parent 0b76e5b4c3
commit 5f134b2a3f
2 changed files with 26 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
openbb_terminal/dashboards/voila/forecast.ipynb vendored
View File

@@ -55,6 +55,7 @@
"from IPython.display import display\n",
"\n",
"from openbb_terminal import api\n",
"openbb = api.openbb\n",
"\n",
"%matplotlib inline\n",
"matplotlib_inline.backend_inline.set_matplotlib_formats(\"svg\")\n",
@@ -71,29 +72,31 @@
"try:\n",
" has_forecast = True\n",
" model_opts = {\n",
" \"expo\": api.forecast.expo,\n",
" \"theta\": api.forecast.theta,\n",
" \"linregr\": api.forecast.linregr,\n",
" \"regr\": api.forecast.regr,\n",
" \"rnn\": api.forecast.rnn,\n",
" \"brnn\": api.forecast.brnn,\n",
" \"nbeats\": api.forecast.nbeats,\n",
" \"tcn\": api.forecast.tcn,\n",
" \"trans\": api.forecast.trans,\n",
" \"tft\": api.forecast.tft,\n",
" \"expo\": openbb.forecast.expo,\n",
" \"theta\": openbb.forecast.theta,\n",
" \"linregr\": openbb.forecast.linregr,\n",
" \"regr\": openbb.forecast.regr,\n",
" \"rnn\": openbb.forecast.rnn,\n",
" \"brnn\": openbb.forecast.brnn,\n",
" \"nbeats\": openbb.forecast.nbeats,\n",
" \"tcn\": openbb.forecast.tcn,\n",
" \"trans\": openbb.forecast.trans,\n",
" \"tft\": openbb.forecast.tft,\n",
" \"nhits\": openbb.forecast.nhits\n",
" }\n",
"\n",
" feat_engs = {\n",
" \"ema\": api.forecast.ema,\n",
" \"sto\": api.forecast.sto,\n",
" \"rsi\": api.forecast.rsi,\n",
" \"roc\": api.forecast.roc,\n",
" \"mom\": api.forecast.mom,\n",
" \"atr\": api.forecast.atr,\n",
" \"delta\": api.forecast.delta,\n",
" \"signal\": api.forecast.signal,\n",
" \"ema\": openbb.forecast.ema,\n",
" \"sto\": openbb.forecast.sto,\n",
" \"rsi\": openbb.forecast.rsi,\n",
" \"roc\": openbb.forecast.roc,\n",
" \"mom\": openbb.forecast.mom,\n",
" \"atr\": openbb.forecast.atr,\n",
" \"delta\": openbb.forecast.delta,\n",
" \"signal\": openbb.forecast.signal,\n",
" }\n",
"except AttributeError:\n",
"except AttributeError as e:\n",
" print(e)\n",
" has_forecast = False\n",
" model_opts = {}\n",
" feat_engs = {}\n",
@@ -207,7 +210,7 @@
" result = calcs\n",
" if not target_column:\n",
" target_column = self.df.columns[0]\n",
" kwargs = {}\n",
" kwargs = {\"chart\": True}\n",
" if contains_covariates and past_covariates != ():\n",
" kwargs[\"past_covariates\"] = \",\".join(past_covariates)\n",
" if has_parameter(forecast_model, \"naive\"):\n",

6
openbb_terminal/dashboards/voila/vsurf.ipynb vendored
View File

@@ -22,7 +22,7 @@
"if module_path not in sys.path:\n",
" sys.path.append(module_path)\n",
"\n",
"from openbb_terminal import api as openbb"
"from openbb_terminal.api import openbb"
]
},
{
@@ -73,7 +73,7 @@
" def create(self, z, ticker, height, width):\n",
" if ticker:\n",
" if ticker != self.last_ticker:\n",
" self.df = openbb.stocks.options.models.yfinance.get_iv_surface(ticker)\n",
" self.df = openbb.stocks.options.vsurf(ticker)\n",
" self.last_ticker = ticker\n",
"\n",
" if not self.df.empty:\n",
@@ -154,7 +154,7 @@
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.8.13"
"version": "3.9.6"
}
},
"nbformat": 4,