Update app.py
Browse files
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['
|
| 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[['
|
| 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='
|
| 1108 |
title="Portfolio Evolution Following Strategy A (Sentiment)",
|
| 1109 |
-
labels={'StrategyA_Cumulative':'Cumulative Value','
|
| 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][['
|
| 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='
|
| 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)"
|