SelmaNajih001 commited on
Commit
daec43f
·
verified ·
1 Parent(s): 9abe4c9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -1026,7 +1026,7 @@ sentiment_pipeline = pipeline("sentiment-analysis", model=MODEL_SENTIMENT)
1026
 
1027
  # --- LOAD DATASET ---
1028
  df_multi = pd.DataFrame(load_dataset(HF_DATASET)["train"])
1029
- df_multi['date'] = pd.to_datetime(df_multi['Data'], errors='coerce')
1030
  df_multi['date_merge'] = df_multi['date'].dt.normalize()
1031
  df_multi.sort_values('date', inplace=True)
1032
 
@@ -1101,12 +1101,12 @@ for company in companies:
1101
  dfs_final[company] = df_c
1102
 
1103
  # --- PLOT STRATEGIES ---
1104
- df_plot_A = pd.concat([df[['Data','StrategyA_Cumulative']].assign(Company=company) for company, df in dfs_final.items()], ignore_index=True)
1105
  #df_plot_B = pd.concat([df[['Data','StrategyB_Cumulative']].assign(Company=company) for company, df in dfs_final.items()], ignore_index=True)
1106
 
1107
- fig_strategy_A = px.line(df_plot_A, x='Data', y='StrategyA_Cumulative', color='Company',
1108
  title="Portfolio Evolution Following Strategy A (Sentiment)",
1109
- labels={'StrategyA_Cumulative':'Cumulative Value','Data':'Date'})
1110
  #fig_strategy_B = px.line(df_plot_B, x='Data', y='StrategyB_Cumulative', color='Company',
1111
  # title="Portfolio Evolution Following Strategy B (Regression)",
1112
  # labels={'StrategyB_Cumulative':'Cumulative Value','Data':'Date'})
@@ -1125,14 +1125,14 @@ def show_company_data(selected_companies):
1125
 
1126
  # --- Strategy plots ---
1127
  df_plot_A = pd.concat(
1128
- [dfs_final[c][['Data','StrategyA_Cumulative']].assign(Company=c) for c in selected_companies if c in dfs_final],
1129
  ignore_index=True
1130
  )
1131
  # df_plot_B = pd.concat([dfs_final[c][['Data','StrategyB_Cumulative']].assign(Company=c) for c in selected_companies if c in dfs_final], ignore_index=True)
1132
 
1133
  fig_A = px.line(
1134
  df_plot_A,
1135
- x='Data',
1136
  y='StrategyA_Cumulative',
1137
  color='Company',
1138
  title="Portfolio Evolution - Strategy A (Sentiment)"
 
1026
 
1027
  # --- LOAD DATASET ---
1028
  df_multi = pd.DataFrame(load_dataset(HF_DATASET)["train"])
1029
+ df_multi['date'] = pd.to_datetime(df_multi['Date'], errors='coerce')
1030
  df_multi['date_merge'] = df_multi['date'].dt.normalize()
1031
  df_multi.sort_values('date', inplace=True)
1032
 
 
1101
  dfs_final[company] = df_c
1102
 
1103
  # --- PLOT STRATEGIES ---
1104
+ df_plot_A = pd.concat([df[['Date','StrategyA_Cumulative']].assign(Company=company) for company, df in dfs_final.items()], ignore_index=True)
1105
  #df_plot_B = pd.concat([df[['Data','StrategyB_Cumulative']].assign(Company=company) for company, df in dfs_final.items()], ignore_index=True)
1106
 
1107
+ fig_strategy_A = px.line(df_plot_A, x='Date', y='StrategyA_Cumulative', color='Company',
1108
  title="Portfolio Evolution Following Strategy A (Sentiment)",
1109
+ labels={'StrategyA_Cumulative':'Cumulative Value','Date':'Date'})
1110
  #fig_strategy_B = px.line(df_plot_B, x='Data', y='StrategyB_Cumulative', color='Company',
1111
  # title="Portfolio Evolution Following Strategy B (Regression)",
1112
  # labels={'StrategyB_Cumulative':'Cumulative Value','Data':'Date'})
 
1125
 
1126
  # --- Strategy plots ---
1127
  df_plot_A = pd.concat(
1128
+ [dfs_final[c][['Date','StrategyA_Cumulative']].assign(Company=c) for c in selected_companies if c in dfs_final],
1129
  ignore_index=True
1130
  )
1131
  # df_plot_B = pd.concat([dfs_final[c][['Data','StrategyB_Cumulative']].assign(Company=c) for c in selected_companies if c in dfs_final], ignore_index=True)
1132
 
1133
  fig_A = px.line(
1134
  df_plot_A,
1135
+ x='Date',
1136
  y='StrategyA_Cumulative',
1137
  color='Company',
1138
  title="Portfolio Evolution - Strategy A (Sentiment)"