virgo-modules 0.2.2__py3-none-any.whl → 0.2.4__py3-none-any.whl

virgo_modules/src/re_utils.py

@@ -760,7 +760,10 @@ def get_data(ticker_name:str, ticket_settings:dict, n_days:int = False, hmm_avai
             object_stock.cluster_hmm_analysis( n_clusters = ticket_settings['settings']['hmm']['n_clusters'],
                                             features_hmm = ticket_settings['settings']['hmm']['features_hmm'],
                                             test_data_size = ticket_settings['settings']['hmm']['test_data_size'],
-                                            seed = ticket_settings['settings']['hmm']['seed'])
+                                            seed = ticket_settings['settings']['hmm']['seed'],
+                                            corr_threshold = ticket_settings['settings']['hmm'].get('corr_threshold',0.75),
+                                            lag_returns_state = ticket_settings['settings']['hmm'].get('lag_returns_state',7),
+                                            )
     
     return object_stock
 

virgo_modules/src/ticketer_source.py

@@ -173,6 +173,8 @@ class stock_eda_panel(object):
         perform additional asset ROC feature, then a new feature is created in the main dataframe
     produce_order_features(feature_name=str, save_features=boolean):
         perform a feature that captures high and low values in an index. this is usefull to know duration/persistence of a signal
+    compute_last_signal (feature_name=str, save_features=boolean):
+        perform a feature that captures high and low values in an index. this is usefull to know duration/persistence of a signal
     create_hmm_derived_features():
         create features derived from hmm states features. Features are the index of the state, the duration of the state, chain raturn
     cluster_hmm_analysis(n_clusters=int,features_hmm=list, test_data_size=int, seed=int, lag_returns_state=int, plot=boolean, save_features=boolean, model=obj):
@@ -1383,6 +1385,53 @@ class stock_eda_panel(object):
             self.signals.append(signal_feature_name)
             self.signals.append(order_feature_name)
 
+    def compute_last_signal(self,feature, save_features = False):
+        """
+        perform two new features when signal is observed, one for the last duration of the previous chain, second for the last duration of the same sign signal
+
+        Parameters
+        ----------
+        feature_name (str): name of the feature
+        save_features (boolean): True to save feature configuration and feature names
+        
+        Returns
+        -------
+        None
+        """
+        def create_last_signal(df, feature, prefix, type ='0'):
+            if type == '0':
+                condition = df[f'order_signal_{feature}'] != 0
+            elif type == '+':
+                condition = df[f'order_signal_{feature}'] > 0
+            elif type == '-':
+                condition = df[f'order_signal_{feature}'] < 0
+            df[f'last_maxorder_{feature}'] = np.where(condition, df[f'order_signal_{feature}'],np.nan)
+            df['tmp_chain_index'] = df[f'last_maxorder_{feature}'].shift(-1)
+            df['last'] = np.where((df[f'last_maxorder_{feature}'] != 0) & (df['tmp_chain_index'].isna()),df[f'last_maxorder_{feature}'], np.nan )
+            df['last'] = df['last'].shift(1)
+            df[f'last_maxorder_{feature}'] = df['last'].fillna(method = 'ffill')
+            df = df.drop(columns = ['tmp_chain_index','last'])
+            df[f'last_maxorder_{feature}'] = np.where(df[f'order_signal_{feature}'] != 0,df[f'last_maxorder_{feature}'],np.nan)
+            df[f'last_maxorder_{feature}'] = df[f'last_maxorder_{feature}'].fillna(0)
+            df = df.rename(columns = {f'last_maxorder_{feature}':f'{prefix}_{feature}'})
+            return df
+        prefix0, prefix1, prefix2 = 'ldur', 'pos', 'neg'
+        self.df = create_last_signal(self.df, feature, prefix0, type ='0')
+        self.df = create_last_signal(self.df, feature, prefix1, type ='+')
+        self.df = create_last_signal(self.df, feature, prefix2, type ='-')
+
+        self.df[f'sldur_{feature}'] = np.where(
+            self.df[f'order_signal_{feature}'] > 0, self.df[f'{prefix1}_{feature}'],
+            np.where(
+                self.df[f'order_signal_{feature}'] < 0, self.df[f'{prefix2}_{feature}'],
+                0
+            )
+        )
+        self.df = self.df.drop(columns = [f'{prefix1}_{feature}',f'{prefix2}_{feature}'])
+        if save_features:
+                self.signals.append(f'sldur_{feature}')
+                self.signals.append(f'ldur_{feature}')
+
     def create_hmm_derived_features(self, lag_returns):
         """
         create features derived from hmm states features. Features are the index of the state, the duration of the state, chain raturn

{virgo_modules-0.2.2.dist-info → virgo_modules-0.2.4.dist-info}/METADATA

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

{virgo_modules-0.2.2.dist-info → virgo_modules-0.2.4.dist-info}/RECORD

@@ -5,11 +5,11 @@ virgo_modules/src/backtester.py,sha256=OhiWyzDX0PthXGuhChyWUmDN3cLkzVYe95zS4nGti
 virgo_modules/src/edge_utils.py,sha256=XN2oEOwADXF9IGNUGx0Ai8B1yDAiU2WDateLnEJh5FE,14243
 virgo_modules/src/hmm_utils.py,sha256=fFWxmh9q3rjiKRHnxNk9k7O4fDrxVxkmp3pbpLvktjc,21116
 virgo_modules/src/pull_artifacts.py,sha256=5OPrgR7pcMSdpbevDRhf0ebk7g7ZRjff4NpTIIWAKjE,1989
-virgo_modules/src/re_utils.py,sha256=05pSVzGKBybPMFGm2wcbYgkSOZs3bZECLjyHvlPlFjM,72490
-virgo_modules/src/ticketer_source.py,sha256=8YInJJF_OzjWXaAuOjAzdaQrgkemmsIpQSTkCZs5VFA,95918
+virgo_modules/src/re_utils.py,sha256=qkNe0BCT3lxm5zAHTEDdH0Cu8ID3A8j3OB39oGFpuDQ,72790
+virgo_modules/src/ticketer_source.py,sha256=bm1I-a_mg80fgdH0Znx-qg0CTUxtVmkL6PWLTCeBTu8,98617
 virgo_modules/src/transformer_utils.py,sha256=LLwKYZRq5hrPVimnq3taD0Lh-q3Bq21fy1I4Icbnxi8,7677
-virgo_modules-0.2.2.dist-info/LICENSE,sha256=pNgFyCYgmimaw0o6V20JupZLROycAnOA_HDDh1tX2V4,1097
-virgo_modules-0.2.2.dist-info/METADATA,sha256=4jz0IHgWzYOrvqW4UnCgHkZHJQWCD9wWWaEm2du9hRY,1428
-virgo_modules-0.2.2.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
-virgo_modules-0.2.2.dist-info/top_level.txt,sha256=ZjI-qEkDtT-8mFwGAWnXfqPOKEGlIhWRW1es1VyXc60,14
-virgo_modules-0.2.2.dist-info/RECORD,,
+virgo_modules-0.2.4.dist-info/LICENSE,sha256=pNgFyCYgmimaw0o6V20JupZLROycAnOA_HDDh1tX2V4,1097
+virgo_modules-0.2.4.dist-info/METADATA,sha256=jI0VJoMF2uIxH6wE4OIDLcOSTERnAETEqPLJecRhLb0,1428
+virgo_modules-0.2.4.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
+virgo_modules-0.2.4.dist-info/top_level.txt,sha256=ZjI-qEkDtT-8mFwGAWnXfqPOKEGlIhWRW1es1VyXc60,14
+virgo_modules-0.2.4.dist-info/RECORD,,