charieswei
/
TradingAgents
mirror of https://github.com/TauricResearch/TradingAgents.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Added visualization for fine-tuned case 

Enhanced plotting for TradingAgents variants with improved line width and alpha settings.
This commit is contained in:
quanliangliu 2025-12-07 21:59:42 -06:00 committed by GitHub
parent 4f71df50ec
commit b7e36038fe
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
evaluation_long_short/visualize.py
View File

@ -416,7 +416,8 @@ def plot_cumulative_returns_from_results(
'KDJ&RSI': 'KDJRSIStrategy', 'KDJ&RSI': 'KDJRSIStrategy',
'ZMR': 'ZMRStrategy', 'ZMR': 'ZMRStrategy',
'SMA': 'SMAStrategy', 'SMA': 'SMAStrategy',
'TradingAgents': 'TradingAgents' 'TradingAgents': 'TradingAgents',
'TradingAgents_DAPT': 'TradingAgents (DAPT+SFT)'
} }
fig, ax = plt.subplots(figsize=figsize) fig, ax = plt.subplots(figsize=figsize)
@ -441,10 +442,16 @@ def plot_cumulative_returns_from_results(
dates = pd.to_datetime([action['date'] for action in data['actions']]) dates = pd.to_datetime([action['date'] for action in data['actions']])
cumulative_returns = [action['cumulative_return'] for action in data['actions']] cumulative_returns = [action['cumulative_return'] for action in data['actions']]
# Plot # Plot with enhanced styling for TradingAgents variants
linewidth = 2.5 if display_name == 'TradingAgents' else 1.5 if 'TradingAgents' in display_name:
linewidth = 2.5
alpha = 0.95
else:
linewidth = 1.5
alpha = 0.8
ax.plot(dates, cumulative_returns, label=display_name, ax.plot(dates, cumulative_returns, label=display_name,
linewidth=linewidth, alpha=0.9) linewidth=linewidth, alpha=alpha)
except Exception as e: except Exception as e:
print(f"Warning: Failed to load {display_name}: {e}") print(f"Warning: Failed to load {display_name}: {e}")