Deepvest/ProfilingAI
0
1o
2 d��g�! � @ s� d dl Zd dlZd dlmZmZmZ d dlm Z d dl3Z4d dl m Z
d dlZd dlmZ d dlmZmZ d dlmZ e G dd� d��ZG d d5� d6�ZG d d � d �ZdS )
� N)�Dict�List�Tuple)� dataclass)�ProcessPoolExecutor)�datetime� timedelta)�EnhancedBacktestResultc @ sL e Zd ZU dZeeef ed< eeef ed< ej ed< eeef ed< dS )�OptimizationResultu- Résultats de l'optimisation de la stratégie� best_params�performance_metrics�optimization_path�parameter_importanceN)7�__name__�8__module__� __qualname__�__doc__r �str�float�__annotations__�pd� DataFrame� r r �B/Users/sandrine/Deepvest_system/src/analysis/strategy_optimizer.pyr9 s 10 11r12 c 13 @ s� e Zd ZdZddefdd�Zdee dej dej d e14fd15d �Z d e j
dee dej dej d ef16d
d�Zded efdd�Zd e17fdd�Zdd� ZdS )�StrategyOptimizeru) Optimiseur de stratégie utilisant Optuna�d �n_trialsc C s || _ || _d | _d S �N)�backtest_enginer �study)�self�enhanced_backtest_enginer r r r �__init__ s 18zStrategyOptimizer.__init__�symbols�19start_date�end_date�returnc � sT �t jddt jjdd�d��_�jj� ���fdd��jdd � ��� }��� |S )20u) Optimise les paramètres de la stratégie�maximize�strategy_optimization�* )�seed)� direction�21study_name�samplerc s �� | ��� �S r )�22_objective)�trial�r% r r$ r# r r �<lambda>/ s z5StrategyOptimizer.optimize_strategy.<locals>.<lambda>T)r �show_progress_bar) �optuna� create_study�samplers�23TPESamplerr �optimizer � _analyze_optimization_results�_plot_optimization_results)r r# r$ r% �optimization_resultsr r0 r �optimize_strategy s � ��z#StrategyOptimizer.optimize_strategyr/ c � s� �|� ddd�|� ddd�|� ddd�|� dd d24�|� d d d
�|� ddd�|� ddd�d�}| jj||||d�I dH }| �|�}|S )z%Fonction objectif pour l'optimisation�technical_weightg�������?g333333�?�alternative_weightg�������?皙�����?�
social_weight�signal_threshold�333333�?gffffff�?�position_size_factorg �?g @� stop_lossg{�G�z�?� take_profitg� ��Q��?g333333�?)r< r= r? r@ rB rC rD )r# r$ r% �strategy_paramsN)�
suggest_floatr �run_enhanced_backtest� _calculate_optimization_score)r r/ r# r$ r% �params�results�optimization_scorer r r r. < s" � � �25z StrategyOptimizer._objectiverJ c C sJ |j d }t|j d �}|j d | }d| d| ddd| }|S )z$Calcule le score pour l'optimisation� sharpe_ratio� max_drawdown�
annual_returnr> rA � )r �abs)r rJ rL rM �return_risk_ratio�scorer r r rH [ s 26��z/StrategyOptimizer._calculate_optimization_scorec C sT | j j}tjdd� | j jD �dd� | j jD �d�}tj�| j �}t|| j j ||d�S )u( Analyse des résultats de l'optimisationc S � g | ]}|j �qS r �rI ��.0�tr r r �27<listcomp>p � zCStrategyOptimizer._analyze_optimization_results.<locals>.<listcomp>c S rS r )�numberrU r r r rX q rY )�index)r r r
r )28r r r r �trialsr3 �29importance�get_param_importancesr30 �31best_value)r r r
r r r r r8 i s ��z/StrategyOptimizer._analyze_optimization_resultsc C s� t j�d� t jdd�}t �d� tjj�| j � t �32d� t �d� tjj� | j � t �33d� t �d� tjj� | j � t �34d � t �d35� t
jd d � | j jD �| j j�� d
�}tj|d� t jdd� t �36d� t �� t �� dS )u+ Visualisation des résultats d'optimisation�seaborn)� �37 ��figsize�� z Progression de l'optimisation�� u Importance des paramètres�� u Corrélations des paramètres�� c S rS r rT rU r r r rX � rY z@StrategyOptimizer._plot_optimization_results.<locals>.<listcomp>)�columns��data�- ��rotationu Distribution des paramètresN)�plt�style�use�figure�subplotr3 �
visualization�38matplotlib�plot_optimization_historyr �title�plot_param_importances�plot_parallel_coordinater r r\ r �keys�sns�boxplot�xticks� tight_layout�show)r �figrk r r r r9 ~ s* 3940414243444546� 47 z,StrategyOptimizer._plot_optimization_resultsN)r )r r r r �intr" r r r � Timestampr48 r; r3 �Trialr r. r rH r8 r9 r r r r r s4 ���49� ����50� r c @ sT e Zd ZdZedefdd��Zedejdejfdd��Z ededej51fd d52��Z d S ) �BacktestVisualizeru4 Visualisation détaillée des résultats de backtestrJ c C s� t j�d� t jdd�}t �d� d| j �� }t �|j|j � t �53d� t �d� t � | j�}t j
|j|j dd d54d � t �55d � t �d
� tj| jdd� t �56d� t �d� tj| jjddd� t �57d� t �d� t �| �}tj|d� t �58d� t �d� t�| j�}tj|j|j d� t jdd� t �59d � t �d � t�| j�}tj|j|j d� t jdd� t �60d � t �� t �� d S ) u/ Création d'un dashboard de performance completr` )� � rc iK rO zPerformance du portefeuilleiL r �redrA )�color�alpha� DrawdownsiM T)�kdezDistribution des rendementsiN �RdYlGn)�cmap�centeru Évolution des positionsiO rj z Performance par type de signaliP )�x�yrl rm u Métriques de risqueiQ u& Contribution des données alternativesN) ro rp rq rr rs �returns�cumprod�plotr[ �valuesrw r� �_calculate_drawdowns� fill_betweenr{ �histplot�heatmap� positions�T�_calculate_signal_returns�barplotr �Series� risk_metricsr} �alternative_contributionr~ r )rJ r� �cumulative_returns� drawdowns�signal_returnsr� �alt_contributionr r r � create_performance_dashboard� s@ 616263 64656667686970 7172 7374 75 z/BacktestVisualizer.create_performance_dashboardr� r& c C s( d| � � }|�� �� }|| d }|S )zCalcul des drawdownsrO )r� � expanding�max)r� �76cumulative� running_maxr� r r r r� � s z'BacktestVisualizer._calculate_drawdownsc C s@ t �� }| jd �� D ]}| jd |k}| j| �� ||< q |S )z(Calcul des rendements par type de signalr+ )r r �signals�uniquer� �mean)rJ r� � signal_type�maskr r r r� � s77 z,BacktestVisualizer._calculate_signal_returnsN) r r r r � staticmethodr r� r r� r� r r� r r r r r� � s 2r� )�numpy�np�pandasr �typingr r r � dataclassesr r3 �matplotlib.pyplot�pyplotro r` r{ �concurrent.futuresr r r �src.models.backtest_resultsr r78 r r� r r r r �<module> s 79