virgo-modules 0.0.32__py3-none-any.whl → 0.0.42__py3-none-any.whl

virgo_modules/src/re_utils.py

@@ -2,6 +2,7 @@ import matplotlib.pyplot as plt
 import matplotlib.gridspec as gridspec
 import seaborn as sns; sns.set()
 import matplotlib.patheffects as path_effects
+from  matplotlib.dates import DateFormatter
 
 import plotly.express as px
 from plotly.subplots import make_subplots
@@ -403,7 +404,7 @@ def rank_by_return(data, lag_days, top_n = 5):
     
     return result
 
-def get_data(ticker_name:str, ticket_settings:dict, n_days:int = False, hmm_available: object = False) -> object:
+def get_data(ticker_name:str, ticket_settings:dict, n_days:int = False, hmm_available: object = False, data_window:str = '5y') -> object:
     """
     this functions runs the stock_eda_panel
     it is shared between train model and predictions
@@ -416,7 +417,7 @@ def get_data(ticker_name:str, ticket_settings:dict, n_days:int = False, hmm_avai
     returns: stock eda panel
     """
 
-    object_stock = stock_eda_panel(ticker_name , n_days )
+    object_stock = stock_eda_panel(ticker_name , n_days, data_window)
     object_stock.get_data()
 
     # computing features if they exists in the ticketr settings
@@ -898,4 +899,45 @@ class produce_plotly_plots:
         if self.show_plot:
             fig.show()
         if self.save_path and self.save_aws:
-            upload_file_to_aws(bucket = 'VIRGO_BUCKET', key = f'market_plots/{self.ticket_name}/'+result_json_name ,input_path = self.save_path+result_json_name)
+            upload_file_to_aws(bucket = 'VIRGO_BUCKET', key = f'market_plots/{self.ticket_name}/'+result_json_name ,input_path = self.save_path+result_json_name)
+
+def plot_hmm_analysis_logger(data_frame,test_data_size, save_path = False, show_plot = True):
+    
+    df = data_frame
+    df_ = df[['Date','hmm_feature','Close',"chain_return"]].sort_values('Date')
+    fig, axs = plt.subplots(1,2,figsize=(10,4))
+    df__ = df_.iloc[:-test_data_size,]
+    sns.boxplot(data=df__, x="hmm_feature", y="chain_return",ax = axs[0]).set_title('train dist')
+    df__ = df_.iloc[-test_data_size:,]
+    sns.boxplot(data=df__ , x="hmm_feature", y="chain_return",ax = axs[1]).set_title('test dist')
+    if save_path:
+        plt.savefig(save_path) 
+    if not show_plot:
+        plt.close()
+
+def plot_hmm_tsanalysis_logger(data_frame, test_data_size,save_path = False, show_plot = True):
+    
+    df = data_frame
+    df_ = df[['Date','hmm_feature','Close',"chain_return"]].sort_values('Date')
+    states = list(df_['hmm_feature'].unique())
+    states.sort()
+    
+    if test_data_size:
+        df__ = df_.iloc[-test_data_size:,]
+        date_limit = pd.Timestamp(str(df__.Date.min().strftime('%Y-%m-%d')))
+    
+    fig, ax1 = plt.subplots(figsize=(10,4))
+    ax1.plot(df_['Date'],df_["Close"])
+    
+    for state in states:
+        df__ = df_[df_.hmm_feature == state]
+        ax1.scatter(df__['Date'],df__["Close"], label = state)
+    formatter = DateFormatter('%Y-%m-%d')
+    if test_data_size:
+        plt.axvline(x=date_limit, color = 'r')
+    fig.legend()
+    fig.autofmt_xdate()
+    if save_path:
+        plt.savefig(save_path) 
+    if not show_plot:
+        plt.close()

virgo_modules/src/ticketer_source.py

@@ -152,12 +152,13 @@ def states_relevance_score(data, default_benchmark_sd = 0.00003, t_threshold = 2
 
 class stock_eda_panel(object):
     
-    def __init__(self, stock_code, n_days):
+    def __init__(self, stock_code, n_days, data_window = '5y'):
         self.stock_code = stock_code
         self.n_days = n_days
         self.today = datetime.date.today()
         self.features = list()
         self.signals = list()
+        self.data_window = data_window
         
     def augmented_dickey_fuller_statistics(self,time_series, label):
         result = adfuller(time_series.dropna().values)
@@ -168,8 +169,7 @@ class stock_eda_panel(object):
         begin_date_str = begin_date.strftime('%Y-%m-%d')
 
         stock = yf.Ticker(self.stock_code)
-        #df = stock.history(period="max")
-        df = stock.history(period='5y')
+        df = stock.history(period=self.data_window)
         
         df = df.sort_values('Date')
         df.reset_index(inplace=True)
@@ -177,7 +177,12 @@ class stock_eda_panel(object):
         df['Date'] = pd.to_datetime(df['Date'])
         
         df = df[df.Date >= begin_date_str ]
-        self.settings_general = {'n_days':self.n_days, 'begin_date':begin_date_str}
+        self.settings_general = {
+            'n_days':self.n_days,
+            'begin_date':begin_date_str,
+            'data_window': self.data_window,
+            'execution_date': self.today.strftime('%Y-%m-%d')
+        }
         self.df = df
         
         ### cleaning volume
@@ -226,8 +231,6 @@ class stock_eda_panel(object):
         df["lower"] = df['Close_roll_mean'] - df["Close_roll_std"]*2
 
         df = df[df.Date >= begin_date_str ]
-        self.settings_general = {'n_days':self.n_days, 'begin_date':begin_date_str}
-        self.df = df
 
         fig = make_subplots(rows=1, cols=1,vertical_spacing = 0.1,shared_xaxes=True,
                            subplot_titles=(
@@ -562,8 +565,7 @@ class stock_eda_panel(object):
         begin_date_str = begin_date.strftime('%Y-%m-%d')
 
         stock = yf.Ticker(self.pair_symbol)
-        #df = stock.history(period="max")
-        df = stock.history(period='5y')
+        df = stock.history(period=self.data_window)
         df = df.sort_values('Date')
         df.reset_index(inplace=True)
         df['Date'] = pd.to_datetime(df['Date'], format='mixed',utc=True).dt.date

{virgo_modules-0.0.32.dist-info → virgo_modules-0.0.42.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: virgo-modules
-Version: 0.0.32
+Version: 0.0.42
 Summary: data processing and statistical modeling using stock market data
 Home-page: https://github.com/miguelmayhem92/virgo_module
 Author: Miguel Mayhuire

virgo_modules-0.0.42.dist-info/RECORD

@@ -0,0 +1,11 @@
+virgo_modules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+virgo_modules/src/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+virgo_modules/src/aws_utils.py,sha256=zvCV_bfN8o8H3iSD-V_aYHtoKqXRD4Kt_T8HIji23WA,965
+virgo_modules/src/pull_artifacts.py,sha256=5OPrgR7pcMSdpbevDRhf0ebk7g7ZRjff4NpTIIWAKjE,1989
+virgo_modules/src/re_utils.py,sha256=KJSFqxl982OKjJk-qtyJPEB-ZcggNPYo5mloNeWrf2c,44873
+virgo_modules/src/ticketer_source.py,sha256=NiK_uILuqkCM_TdBEetphtYgSjmo7WsGMmsWmAOQazs,94469
+virgo_modules-0.0.42.dist-info/LICENSE,sha256=pNgFyCYgmimaw0o6V20JupZLROycAnOA_HDDh1tX2V4,1097
+virgo_modules-0.0.42.dist-info/METADATA,sha256=BDXW-DRp_lWmbelKSsjZJOg5UPei3_g-OO4LZqsetnE,1429
+virgo_modules-0.0.42.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
+virgo_modules-0.0.42.dist-info/top_level.txt,sha256=ZjI-qEkDtT-8mFwGAWnXfqPOKEGlIhWRW1es1VyXc60,14
+virgo_modules-0.0.42.dist-info/RECORD,,

virgo_modules-0.0.32.dist-info/RECORD

@@ -1,11 +0,0 @@
-virgo_modules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-virgo_modules/src/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-virgo_modules/src/aws_utils.py,sha256=zvCV_bfN8o8H3iSD-V_aYHtoKqXRD4Kt_T8HIji23WA,965
-virgo_modules/src/pull_artifacts.py,sha256=5OPrgR7pcMSdpbevDRhf0ebk7g7ZRjff4NpTIIWAKjE,1989
-virgo_modules/src/re_utils.py,sha256=yj44T--q9F-tQWisRkelp_oIce1h3KGE0AjuSJQSZ0U,43251
-virgo_modules/src/ticketer_source.py,sha256=hVT2P6LD_h9GFExQfyNimxPuETsWaC-kuVU_qlomp7I,94431
-virgo_modules-0.0.32.dist-info/LICENSE,sha256=pNgFyCYgmimaw0o6V20JupZLROycAnOA_HDDh1tX2V4,1097
-virgo_modules-0.0.32.dist-info/METADATA,sha256=zRBSYCNq2-M7V-ido3uV6KhX8QaIkYggcdUtJIiavhY,1429
-virgo_modules-0.0.32.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
-virgo_modules-0.0.32.dist-info/top_level.txt,sha256=ZjI-qEkDtT-8mFwGAWnXfqPOKEGlIhWRW1es1VyXc60,14
-virgo_modules-0.0.32.dist-info/RECORD,,