Added export to stocks/qa/beta (#2510)

* added export to beta

* added alpha and beta to filename and removed it from the actual file
This commit is contained in:
hjoaquim
2022-09-14 15:16:33 +01:00
committed by GitHub
parent 271d677f2c
commit d508c5e815
2 changed files with 15 additions and 1 deletions

View File

@@ -2,12 +2,14 @@
__docformat__ = "numpy"
import logging
import os
import matplotlib.pyplot as plt
import pandas as pd
from openbb_terminal.decorators import log_start_end
from openbb_terminal.rich_config import console
from openbb_terminal.stocks.quantitative_analysis.beta_model import beta_model
from openbb_terminal.helper_funcs import export_data
logger = logging.getLogger(__name__)
@@ -18,6 +20,7 @@ def beta_view(
ref_symbol: str,
data: pd.DataFrame = None,
ref_data: pd.DataFrame = None,
export: str = "",
) -> None:
"""Display the beta scatterplot + linear regression.
@@ -52,3 +55,12 @@ def beta_view(
ax.text(0.9, 0.1, beta_text, horizontalalignment="right", transform=ax.transAxes)
fig.show()
console.print()
df = pd.DataFrame({"sr": sr, "rr": rr})
export_data(
export,
os.path.dirname(os.path.abspath(__file__)),
f"beta_alpha={alpha}_beta={beta}",
df,
)

View File

@@ -783,7 +783,9 @@ class QaController(StockBaseController):
parser, other_args, EXPORT_ONLY_RAW_DATA_ALLOWED
)
if ns_parser:
beta_view(self.ticker, ns_parser.ref, data=self.stock)
beta_view(
self.ticker, ns_parser.ref, data=self.stock, export=ns_parser.export
)
@log_start_end(log=logger)
def call_var(self, other_args: List[str]):