virgo-modules 0.0.71__py3-none-any.whl → 0.0.72__py3-none-any.whl

virgo_modules/src/ticketer_source.py

@@ -1217,7 +1217,20 @@ class stock_eda_panel(object):
             ### first feature: the hidden state
             self.df['hmm_feature'] = self.model_hmm.predict(self.df)
             self.create_hmm_derived_features(lag_returns = lag_returns_state)
-        
+
+            ## completion
+
+            hidden_states = pipeline_hmm.predict(data_train)
+            map_ = {i:f'state_{i}' for i in range(n_clusters)}
+            color_map = { i:DEFAULT_PLOTLY_COLORS[i] for i in range(n_clusters)}
+
+            data_train['HMM'] = hidden_states
+            data_train['HMM_state'] =  data_train['HMM'].map(map_)
+
+            hidden_states = pipeline_hmm.predict(data_test)
+            data_test['HMM'] = hidden_states
+            data_test['HMM_state'] =  data_test['HMM'].map(map_)
+            
         if model:
             self.df['hmm_feature'] = model.predict(self.df)
             self.create_hmm_derived_features(lag_returns = lag_returns_state)
@@ -1229,13 +1242,6 @@ class stock_eda_panel(object):
 
         if plot:
 
-            hidden_states = pipeline_hmm.predict(data_train)
-            map_ = {i:f'state_{i}' for i in range(n_clusters)}
-            color_map = { i:DEFAULT_PLOTLY_COLORS[i] for i in range(n_clusters)}
-
-            data_train['HMM'] = hidden_states
-            data_train['HMM_state'] =  data_train['HMM'].map(map_)
-
             fig = go.Figure()
             fig.add_trace(go.Scatter(x=data_train['Date'], y=data_train['Close'], mode='lines',name = 'history', marker_color = 'grey'))
             for state in data_train['HMM_state'].unique():
@@ -1246,10 +1252,6 @@ class stock_eda_panel(object):
             fig.show()
 
             print('---------------------------------------------------------')
-        
-            hidden_states = pipeline_hmm.predict(data_test)
-            data_test['HMM'] = hidden_states
-            data_test['HMM_state'] =  data_test['HMM'].map(map_)
 
             fig = go.Figure()
             fig.add_trace(go.Scatter(x=data_test['Date'], y=data_test['Close'], mode='lines',name = 'history', marker_color = 'grey'))

{virgo_modules-0.0.71.dist-info → virgo_modules-0.0.72.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: virgo-modules
-Version: 0.0.71
+Version: 0.0.72
 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.71.dist-info → virgo_modules-0.0.72.dist-info}/RECORD

@@ -4,9 +4,9 @@ virgo_modules/src/aws_utils.py,sha256=GWmVdXM0mIJJPn-X-bEtM4KtNPCHM1D457hnuKxaM7
 virgo_modules/src/edge_utils.py,sha256=Ihdmq7dyb8gOvG6CrDal7wsa15tqsdsFk6KINwM6578,7691
 virgo_modules/src/pull_artifacts.py,sha256=5OPrgR7pcMSdpbevDRhf0ebk7g7ZRjff4NpTIIWAKjE,1989
 virgo_modules/src/re_utils.py,sha256=LDI3sYAaNm3LO5gRul7PyCVbJrkT3PBihObkdVilVec,52428
-virgo_modules/src/ticketer_source.py,sha256=fGj954x4UmYU4zaI20q5dTkgqAVK0XQJsuibM2UBPMs,104812
-virgo_modules-0.0.71.dist-info/LICENSE,sha256=pNgFyCYgmimaw0o6V20JupZLROycAnOA_HDDh1tX2V4,1097
-virgo_modules-0.0.71.dist-info/METADATA,sha256=vWlzxe3a8nehV54nuJZi87OWDKwwEhz5JNRkncIv_To,1429
-virgo_modules-0.0.71.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
-virgo_modules-0.0.71.dist-info/top_level.txt,sha256=ZjI-qEkDtT-8mFwGAWnXfqPOKEGlIhWRW1es1VyXc60,14
-virgo_modules-0.0.71.dist-info/RECORD,,
+virgo_modules/src/ticketer_source.py,sha256=ciMPObqntAFtnlY1IPt8-Y4mz6yuD1jy6gRQN109D4M,104837
+virgo_modules-0.0.72.dist-info/LICENSE,sha256=pNgFyCYgmimaw0o6V20JupZLROycAnOA_HDDh1tX2V4,1097
+virgo_modules-0.0.72.dist-info/METADATA,sha256=Txin9qouILtGSvPTQYcJPPkWXNry0JjI3sSfAMB0Cjg,1429
+virgo_modules-0.0.72.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
+virgo_modules-0.0.72.dist-info/top_level.txt,sha256=ZjI-qEkDtT-8mFwGAWnXfqPOKEGlIhWRW1es1VyXc60,14
+virgo_modules-0.0.72.dist-info/RECORD,,