File size: 934 Bytes
0c90d5f
4e938bd
b57f75f
 
0c90d5f
b57f75f
0c90d5f
b57f75f
0c90d5f
 
 
b57f75f
 
 
 
 
 
 
 
 
 
0c90d5f
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
from .fortitran import BaseFortiTranEstimator
from src.config.schemas import SystemConfig, ModelConfig


class AdaFortiTranEstimator(BaseFortiTranEstimator):
    """
    Adaptive Hybrid CNN-Transformer Channel Estimator for OFDM Systems with channel adaptation.

    This model extends the base estimator with channel adaptation capabilities,
    incorporating channel conditions (SNR, delay spread, Doppler shift) into
    the estimation process through conditional attention mechanisms.
    """

    def __init__(self, system_config: SystemConfig, model_config: ModelConfig) -> None:
        """
        Initialize the AdaFortiTranEstimator.

        Args:
            system_config: OFDM system configuration (subcarriers, symbols, pilot arrangement)
            model_config: Model architecture configuration (patch size, layers, etc.)
        """
        super().__init__(system_config, model_config, use_channel_adaptation=True)