mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-29 13:32:08 +00:00
97 lines
3.0 KiB
Python
97 lines
3.0 KiB
Python
"""Monte Carlo View"""
|
|
__docformat__ = "numpy"
|
|
|
|
import logging
|
|
import os
|
|
from typing import List, Optional, Union
|
|
|
|
import matplotlib.pyplot as plt
|
|
import numpy as np
|
|
import pandas as pd
|
|
import seaborn as sns
|
|
|
|
from gamestonk_terminal.config_terminal import theme
|
|
from gamestonk_terminal.common.prediction_techniques import mc_model
|
|
from gamestonk_terminal.config_plot import PLOT_DPI
|
|
from gamestonk_terminal.decorators import log_start_end
|
|
from gamestonk_terminal.helper_funcs import (
|
|
export_data,
|
|
get_next_stock_market_days,
|
|
plot_autoscale,
|
|
)
|
|
from gamestonk_terminal.rich_config import console
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
@log_start_end(log=logger)
|
|
def display_mc_forecast(
|
|
data: Union[pd.DataFrame, pd.Series],
|
|
n_future: int,
|
|
n_sims: int,
|
|
use_log=True,
|
|
fig_title: str = "",
|
|
export: str = "",
|
|
time_res: str = "",
|
|
external_axes: Optional[List[plt.Axes]] = None,
|
|
):
|
|
"""Display monte carlo forecasting
|
|
|
|
Parameters
|
|
----------
|
|
data : Union[pd.Series, np.array]
|
|
Data to forecast
|
|
n_future : int
|
|
Number of days to forecast
|
|
n_sims : int
|
|
Number of simulations to run
|
|
use_log : bool, optional
|
|
Flag to use lognormal, by default True
|
|
fig_title : str
|
|
Figure title
|
|
export: str
|
|
Format to export data
|
|
time_res : str
|
|
Resolution for data, allowing for predicting outside of standard market days
|
|
external_axes : Optional[List[plt.Axes]], optional
|
|
External axes (2 axis is expected in the list), by default None
|
|
"""
|
|
predicted_values = mc_model.get_mc_brownian(data, n_future, n_sims, use_log)
|
|
if not time_res or time_res == "1D":
|
|
future_index = get_next_stock_market_days(data.index[-1], n_next_days=n_future) # type: ignore
|
|
else:
|
|
future_index = pd.date_range(data.index[-1], periods=n_future + 1, freq=time_res)[1:] # type: ignore
|
|
|
|
# This plot has 1 axis
|
|
if external_axes is None:
|
|
_, (ax1, ax2) = plt.subplots(1, 2, figsize=plot_autoscale(), dpi=PLOT_DPI)
|
|
else:
|
|
if len(external_axes) != 2:
|
|
logger.error("Expected list of two axis items")
|
|
console.print("[red]Expected list of two axis items./n[/red]")
|
|
return
|
|
(ax1, ax2) = external_axes
|
|
|
|
ax1.plot(data)
|
|
ax1.plot(future_index, predicted_values, alpha=0.3)
|
|
start_timestamp = data.index[0]
|
|
end_timestamp = future_index[-1]
|
|
ax1.set_xlim(start_timestamp, end_timestamp)
|
|
ax1.set_title(f"{fig_title} Data Predictions")
|
|
|
|
sns.histplot(predicted_values[-1, :], ax=ax2, kde=True)
|
|
ax2.set_xlabel("Final Value")
|
|
ax2.axvline(
|
|
x=data.values[-1], color=theme.down_color, label="Last Value", linestyle="-"
|
|
)
|
|
ax2.set_title(f"Distribution of final values after {n_future} steps.")
|
|
ax2.set_xlim(np.min(predicted_values[-1, :]), np.max(predicted_values[-1, :]))
|
|
ax2.legend()
|
|
|
|
theme.style_primary_axis(ax1)
|
|
theme.style_primary_axis(ax2)
|
|
if external_axes is None:
|
|
theme.visualize_output()
|
|
|
|
export_data(export, os.path.dirname(os.path.abspath(__file__)), "mc")
|