Local changes: Updated model training, removed debug instrumentation, and configuration improvements
This commit is contained in:
144
src/risk/position_sizing.py
Normal file
144
src/risk/position_sizing.py
Normal file
@@ -0,0 +1,144 @@
|
||||
"""Position sizing rules."""
|
||||
|
||||
from decimal import Decimal
|
||||
from typing import Optional
|
||||
from src.core.config import get_config
|
||||
from src.core.logger import get_logger
|
||||
from src.exchanges.base import BaseExchangeAdapter
|
||||
|
||||
logger = get_logger(__name__)
|
||||
|
||||
|
||||
class PositionSizingManager:
|
||||
"""Manages position sizing calculations."""
|
||||
|
||||
def __init__(self):
|
||||
"""Initialize position sizing manager."""
|
||||
self.config = get_config()
|
||||
self.logger = get_logger(__name__)
|
||||
|
||||
def calculate_size(
|
||||
self,
|
||||
symbol: str,
|
||||
price: Decimal,
|
||||
balance: Decimal,
|
||||
risk_percent: Optional[Decimal] = None,
|
||||
exchange_adapter: Optional[BaseExchangeAdapter] = None
|
||||
) -> Decimal:
|
||||
"""Calculate position size, accounting for fees.
|
||||
|
||||
Args:
|
||||
symbol: Trading symbol
|
||||
price: Entry price
|
||||
balance: Available balance
|
||||
risk_percent: Risk percentage (uses config default if None)
|
||||
exchange_adapter: Exchange adapter for fee calculation (optional)
|
||||
|
||||
Returns:
|
||||
Calculated position size
|
||||
"""
|
||||
if risk_percent is None:
|
||||
risk_percent = Decimal(str(
|
||||
self.config.get("risk.position_size_percent", 2.0)
|
||||
)) / 100
|
||||
|
||||
position_value = balance * risk_percent
|
||||
|
||||
# Account for fees by reserving fee amount
|
||||
from src.trading.fee_calculator import get_fee_calculator
|
||||
fee_calculator = get_fee_calculator()
|
||||
|
||||
# Reserve ~0.4% for round-trip fees
|
||||
fee_reserve = fee_calculator.calculate_fee_reserve(
|
||||
position_value=position_value,
|
||||
exchange_adapter=exchange_adapter,
|
||||
reserve_percent=0.004 # 0.4% for round-trip
|
||||
)
|
||||
|
||||
# Adjust position value to account for fees
|
||||
adjusted_position_value = position_value - fee_reserve
|
||||
|
||||
if price > 0:
|
||||
quantity = adjusted_position_value / price
|
||||
return max(Decimal(0), quantity) # Ensure non-negative
|
||||
|
||||
return Decimal(0)
|
||||
|
||||
def calculate_kelly_criterion(
|
||||
self,
|
||||
win_rate: float,
|
||||
avg_win: float,
|
||||
avg_loss: float
|
||||
) -> Decimal:
|
||||
"""Calculate position size using Kelly Criterion.
|
||||
|
||||
Args:
|
||||
win_rate: Win rate (0.0 to 1.0)
|
||||
avg_win: Average win amount
|
||||
avg_loss: Average loss amount
|
||||
|
||||
Returns:
|
||||
Kelly percentage
|
||||
"""
|
||||
if avg_loss == 0:
|
||||
return Decimal(0)
|
||||
|
||||
kelly = (win_rate * avg_win - (1 - win_rate) * avg_loss) / avg_win
|
||||
# Use fractional Kelly (half) for safety
|
||||
return Decimal(str(kelly / 2))
|
||||
|
||||
def validate_position_size(
|
||||
self,
|
||||
symbol: str,
|
||||
quantity: Decimal,
|
||||
price: Decimal,
|
||||
balance: Decimal,
|
||||
exchange_adapter: Optional[BaseExchangeAdapter] = None
|
||||
) -> bool:
|
||||
"""Validate position size against limits, accounting for fees.
|
||||
|
||||
Args:
|
||||
symbol: Trading symbol
|
||||
quantity: Position quantity
|
||||
price: Entry price
|
||||
balance: Available balance
|
||||
exchange_adapter: Exchange adapter for fee calculation (optional)
|
||||
|
||||
Returns:
|
||||
True if position size is valid
|
||||
"""
|
||||
position_value = quantity * price
|
||||
|
||||
# Calculate estimated fee for this trade
|
||||
from src.trading.fee_calculator import get_fee_calculator
|
||||
from src.core.database import OrderType
|
||||
|
||||
fee_calculator = get_fee_calculator()
|
||||
estimated_fee = fee_calculator.calculate_fee(
|
||||
quantity=quantity,
|
||||
price=price,
|
||||
order_type=OrderType.MARKET, # Use market as worst case
|
||||
exchange_adapter=exchange_adapter
|
||||
)
|
||||
|
||||
total_cost = position_value + estimated_fee
|
||||
|
||||
# Check if exceeds available balance (including fees)
|
||||
if total_cost > balance:
|
||||
self.logger.warning(
|
||||
f"Position cost {total_cost} (value: {position_value}, fee: {estimated_fee}) "
|
||||
f"exceeds balance {balance}"
|
||||
)
|
||||
return False
|
||||
|
||||
# Check against risk limits
|
||||
max_position_percent = Decimal(str(
|
||||
self.config.get("risk.position_size_percent", 2.0)
|
||||
)) / 100
|
||||
|
||||
if position_value > balance * max_position_percent:
|
||||
self.logger.warning(f"Position size exceeds risk limit")
|
||||
return False
|
||||
|
||||
return True
|
||||
|
||||
Reference in New Issue
Block a user