virgo-modules 0.8.1__py3-none-any.whl → 0.8.2__py3-none-any.whl

virgo_modules/src/edge_utils/edge_utils.py

@@ -450,7 +450,7 @@ def edge_probas_lines(data, threshold, plot = False, look_back = 750):
         fig.show()
     return fig
 
-def get_rolling_probs(data, window = 3,plot = False, look_back = 750):
+def get_rolling_probs(data, window = 3,plot = False, look_back = 750, rets_eval=7):
     """
     produce a plotly plot of smoothed edges and closing prices
 
@@ -465,6 +465,7 @@ def get_rolling_probs(data, window = 3,plot = False, look_back = 750):
     """
     prob_cols = ['proba_target_down','proba_target_up']
     df = data[prob_cols+['Date','log_return','Close']].iloc[-look_back:]
+    df["eval_rets"] = (df["Close"]/df["Close"].shift(rets_eval)-1)*100
     for colx in prob_cols:
         df[f'roll_{colx}'] = df.sort_values('Date')[colx].rolling(window, min_periods=1).mean()
     df['roll_edge'] = np.where(df['roll_proba_target_up'] > df['roll_proba_target_down'],'up','down')
@@ -480,14 +481,14 @@ def get_rolling_probs(data, window = 3,plot = False, look_back = 750):
     fig = make_subplots(
         rows=2, cols=2,shared_xaxes=False,vertical_spacing=0.08,
         specs=[[{"colspan": 2, "secondary_y":True}, None],[{}, {}]],
-            subplot_titles=("Smooth edge probabilities", "expected log return", "Duration"))
+            subplot_titles=("Smooth edge probabilities", f"expected return {rets_eval} days", "Duration"))
     fig.add_trace(go.Scatter(x=df.Date, y=df.Close,mode='lines+markers',name='Close price'))
     fig.add_trace(go.Scatter(x=df.Date, y=df.roll_proba_target_down,mode='lines',marker = dict(color = 'coral'),name='go down'),secondary_y=True,col=1,row=1)
     fig.add_trace(go.Scatter(x=df.Date, y=df.roll_proba_target_up,mode='lines',marker = dict(opacity=0.1,size=80), name='go up'),secondary_y=True,col=1,row=1)
 
     for re in df['roll_edge'].unique():
-         fig.add_trace(go.Box(x=df[df['roll_edge']==re]["log_return"],name=re,marker_color=colors.get(re),showlegend=False),col=1,row=2)
-
+         fig.add_trace(go.Box(x=df[df['roll_edge']==re]["eval_rets"],name=re,marker_color=colors.get(re),showlegend=False),col=1,row=2)
+         fig.add_vline(x=0, line_width=2, line_dash="dash", line_color="grey", col=1,row=2)
     df_ = df.groupby(['roll_edge','chain'],as_index=False).agg(max_duration = ('chain_id','max'))
     for re in df_['roll_edge'].unique():
         fig.add_trace(go.Box(x=df_[df_['roll_edge']==re]["max_duration"],name=re,marker_color=colors.get(re),showlegend=False),col=2,row=2)

{virgo_modules-0.8.1.dist-info → virgo_modules-0.8.2.dist-info}/METADATA

@@ -1,12 +1,11 @@
-Metadata-Version: 2.1
-Name: virgo-modules
-Version: 0.8.1
+Metadata-Version: 2.4
+Name: virgo_modules
+Version: 0.8.2
 Summary: data processing and statistical modeling using stock market data
 Home-page: https://github.com/miguelmayhem92/virgo_module
 Author: Miguel Mayhuire
 Author-email: miguelmayhem92@gmail.com
 License: MIT
-Platform: UNKNOWN
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Operating System :: OS Independent
@@ -14,7 +13,18 @@ Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Provides-Extra: dev
-Requires-Dist: pytest (>=7.0) ; extra == 'dev'
+Requires-Dist: pytest>=7.0; extra == "dev"
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: license
+Dynamic: license-file
+Dynamic: provides-extra
+Dynamic: requires-python
+Dynamic: summary
 
 # Virgo Package
 
@@ -35,4 +45,3 @@ obj = stock_eda_panel(stock_code = 'PEP', n_days = 20)
 obj.get_data()
 print(obj.df.shape)
 ```
-

{virgo_modules-0.8.1.dist-info → virgo_modules-0.8.2.dist-info}/RECORD

@@ -9,14 +9,14 @@ virgo_modules/src/ticketer_source.py,sha256=528WhGoANOm4IKnxGSWsbQxxUh3-qlZfvGRN
 virgo_modules/src/transformer_utils.py,sha256=SnYdtsFPnSF6u4UFIat0-X3-qVuUWvv_T46kiB-H0Sk,13682
 virgo_modules/src/edge_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 virgo_modules/src/edge_utils/conformal_utils.py,sha256=cKm4KSM261Eu1FJn4oowKYiKIesW81VbqITIvopGSVk,5410
-virgo_modules/src/edge_utils/edge_utils.py,sha256=4uXVWthzJDzkJ4Uq19ZYL9aPcA6CDUS3xYD4FY-a2AM,20018
+virgo_modules/src/edge_utils/edge_utils.py,sha256=8xdcmeEPyfD76c-mcdfkHqWjwtJOPoUsit5UH3cRJTg,20207
 virgo_modules/src/edge_utils/feature_selection.py,sha256=HYbQ0JLPDiRYhn-5-C438YEKbuNduDmuvboFC_VkHww,2453
 virgo_modules/src/edge_utils/shap_utils.py,sha256=FgcHkfddvdFSeUqEubYa2ExRGVAWSthqK4b-eKagEmo,2333
 virgo_modules/src/edge_utils/stack_model.py,sha256=QqE91uLo2KauGEj91AVNANB1xE7J4Fa49YOX7k5mFng,4257
 virgo_modules/src/market/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 virgo_modules/src/market/market_tools.py,sha256=vBt66_7E3ANz7avzfeNw_RHMGvG9lh5PRhxmcf_Oyjc,6880
-virgo_modules-0.8.1.dist-info/LICENSE,sha256=pNgFyCYgmimaw0o6V20JupZLROycAnOA_HDDh1tX2V4,1097
-virgo_modules-0.8.1.dist-info/METADATA,sha256=2RrX9eUV6HHFhoW0osSzUJNEmXaWeTg65U1y-mmlC_g,899
-virgo_modules-0.8.1.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
-virgo_modules-0.8.1.dist-info/top_level.txt,sha256=ZjI-qEkDtT-8mFwGAWnXfqPOKEGlIhWRW1es1VyXc60,14
-virgo_modules-0.8.1.dist-info/RECORD,,
+virgo_modules-0.8.2.dist-info/licenses/LICENSE,sha256=pNgFyCYgmimaw0o6V20JupZLROycAnOA_HDDh1tX2V4,1097
+virgo_modules-0.8.2.dist-info/METADATA,sha256=XU1VhD9yFBTwoHukEDwy4TaEZ2_SI2IW4rJstkqAu7M,1122
+virgo_modules-0.8.2.dist-info/WHEEL,sha256=lTU6B6eIfYoiQJTZNc-fyaR6BpL6ehTzU3xGYxn2n8k,91
+virgo_modules-0.8.2.dist-info/top_level.txt,sha256=ZjI-qEkDtT-8mFwGAWnXfqPOKEGlIhWRW1es1VyXc60,14
+virgo_modules-0.8.2.dist-info/RECORD,,

{virgo_modules-0.8.1.dist-info → virgo_modules-0.8.2.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.38.4)
+Generator: setuptools (78.1.1)
 Root-Is-Purelib: true
 Tag: py3-none-any